Update script.py

script.py

@@ -71,8 +71,9 @@ def run_training(hf_dataset_path: str):
     dataset_dir = process_dataset(dataset_dir)
 
     # run training
-    commands = "git clone https://github.com/ostris/ai-toolkit.git ai-toolkit && cd ai-toolkit && git submodule update --init --recursive"
-    subprocess.run(commands, shell=True)
+    if not os.path.exists("ai-toolkit"):
+        commands = "git clone https://github.com/ostris/ai-toolkit.git ai-toolkit && cd ai-toolkit && git submodule update --init --recursive"
+        subprocess.run(commands, shell=True)
 
     commands = f"python run.py {os.path.join(dataset_dir, 'config.yaml')}"
     process = subprocess.Popen(commands, shell=True, cwd="ai-toolkit", env=os.environ)