Doesn't run pip when already installed

2023-01-11 17:55:02 +01:00 · 2023-01-11 17:55:02 +01:00 · 33683c21ea
parent db93f06a21
commit 33683c21ea
1 changed files with 4 additions and 1 deletions
--- a/install.py
+++ b/install.py
@ -1,2 +1,5 @@
 import launch
-launch.run_pip("install --upgrade transformers", "Requirment of Prompt-Maker")
+if not launch.is_installed("transformers"):
+    launch.run_pip("install --upgrade transformers", "Requirement of Prompt-Maker")
+if not launch.is_installed("torch"):
+    launch.run_pip("install --upgrade torch", "Requirement of Prompt-Maker")