From 33683c21eaacbc9386f2ee73c711e42c5b890893 Mon Sep 17 00:00:00 2001 From: Imrayya Date: Wed, 11 Jan 2023 17:55:02 +0100 Subject: [PATCH] Doesn't run pip when already installed --- install.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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