diff --git a/install.py b/install.py index d3106e4..c22dc08 100644 --- a/install.py +++ b/install.py @@ -1,2 +1,5 @@ import launch -launch.run_pip("install --upgrade transformers", "Requirment of Prompt-Maker") \ No newline at end of file +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") \ No newline at end of file