Update install.py

Anshler · Dec 23, 2023 · 3183d41 · 3183d41
1 parent 0a44ca1
commit 3183d41
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/install.py b/install.py
@@ -4,9 +4,9 @@
 
 # if not launch.is_installed("aitextgen"):
 #     launch.run_pip("install aitextgen==0.6.0", "requirements for MagicPrompt")
-if not launch.is_installed("transformers>=4.35.2"):
-    launch.run_pip("install accelerate>=0.20.3", "requirements for Image Caption")
-    launch.run_pip("install transformers>=4.35.2", "requirements for Image Caption")
+if not launch.is_installed("transformers==4.35.2"):
+    launch.run_pip("install accelerate==0.20.3", "requirements for Image Caption")
+    launch.run_pip("install transformers==4.35.2", "requirements for Image Caption")
 
 if not launch.is_installed("open_clip_torch"):
     launch.run_pip("install open_clip_torch", "requirements for Image Caption")