diff --git a/common/python.nix b/common/python.nix
index 547e1e0..6c6e41c 100644
--- a/common/python.nix
+++ b/common/python.nix
@@ -13,6 +13,8 @@
       python312Packages.jax
       python312Packages.huggingface-hub
       python312Packages.datasets
+      python312Packages.transformers
+      python312Packages.diffusers
       python312Packages.jupyter
     ]))
     pyenv