Update script.py

script.py

@@ -124,8 +124,13 @@ def run_training(hf_dataset_path: str):
     # run training
     if not os.path.exists("ai-toolkit"):
         commands = "git clone https://github.com/DevW1ld/ai-toolkit.git ai-toolkit && cd ai-toolkit && git submodule update --init --recursive"
+        shutil.rmtree(os.path.join(toolkit_src, ".git"), ignore_errors=True)
+        shutil.rmtree(os.path.join(toolkit_src, ".gitmodules"), ignore_errors=True)
         subprocess.run(commands, shell=True)
 
+
+
+
     # patch_ai_toolkit_typing()
     commands = f"python run.py {os.path.join(dataset_dir, 'config.yaml')}"
     process = subprocess.Popen(commands, shell=True, cwd="ai-toolkit", env=os.environ, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True,)