version pinning dreambooth (#534)

modal-labs · Jan 3, 2024 · e4f7bf5 · e4f7bf5
1 parent 5f3471e
commit e4f7bf5
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/06_gpu_and_ml/dreambooth/dreambooth_app.py b/06_gpu_and_ml/dreambooth/dreambooth_app.py
@@ -52,13 +52,13 @@
     .pip_install(
         "accelerate==0.19",
         "datasets~=2.13",
-        "ftfy",
+        "ftfy~=6.1",
         "gradio~=3.10",
-        "smart_open",
-        "transformers",
-        "torch",
-        "torchvision",
-        "triton",
+        "smart_open~=6.4",
+        "transformers==4.28",
+        "torch~=2.1",
+        "torchvision~=0.16",
+        "triton~=2.1",
     )
     .pip_install("xformers", pre=True)
     .apt_install("git")