Update script.py

script.py

@@ -64,7 +64,7 @@ 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"
+    commands = "git clone https://github.com/ostris/ai-toolkit.git ai-toolkit && cd ai-toolkit && git checkout bc693488eb3cf48ded8bc2af845059d80f4cf7d0 && git submodule update --init --recursive"
     subprocess.run(commands, shell=True)
 
     commands = f"python run.py {os.path.join(dataset_dir, 'config.yaml')}"