From e4f7bf53293a955139637e1dfa0d5f1be69659ee Mon Sep 17 00:00:00 2001
From: Jonathon Belotti <jonathon@modal.com>
Date: Wed, 3 Jan 2024 18:28:52 +1100
Subject: [PATCH] version pinning dreambooth (#534)

---
 06_gpu_and_ml/dreambooth/dreambooth_app.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/06_gpu_and_ml/dreambooth/dreambooth_app.py b/06_gpu_and_ml/dreambooth/dreambooth_app.py
index 71bec97ef..0a9c11d5c 100644
--- 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")