Commit
•
867b96b
1
Parent(s):
e470146
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ from pathlib import Path
|
|
18 |
import spaces
|
19 |
MAX_IMAGES = 50
|
20 |
|
21 |
-
training_script_url = "https://raw.githubusercontent.com/huggingface/diffusers/
|
22 |
subprocess.run(['wget', '-N', training_script_url])
|
23 |
orchestrator_script_url = "https://huggingface.co/datasets/multimodalart/lora-ease-helper/raw/main/script.py"
|
24 |
subprocess.run(['wget', '-N', orchestrator_script_url])
|
@@ -306,7 +306,7 @@ def start_training(
|
|
306 |
shutil.copy("train_dreambooth_lora_sdxl_advanced.py", f"{spacerunner_folder}/trainer.py")
|
307 |
shutil.copy("script.py", f"{spacerunner_folder}/script.py")
|
308 |
shutil.copytree(dataset_folder, f"{spacerunner_folder}/{dataset_folder}")
|
309 |
-
requirements='''
|
310 |
-huggingface_hub
|
311 |
torch
|
312 |
git+https://github.com/huggingface/diffusers@5b186b712837104b40d095b26ed6a2ec61246cb4
|
|
|
18 |
import spaces
|
19 |
MAX_IMAGES = 50
|
20 |
|
21 |
+
training_script_url = "https://raw.githubusercontent.com/huggingface/diffusers/add-peft-to-advanced-training-script/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py"
|
22 |
subprocess.run(['wget', '-N', training_script_url])
|
23 |
orchestrator_script_url = "https://huggingface.co/datasets/multimodalart/lora-ease-helper/raw/main/script.py"
|
24 |
subprocess.run(['wget', '-N', orchestrator_script_url])
|
|
|
306 |
shutil.copy("train_dreambooth_lora_sdxl_advanced.py", f"{spacerunner_folder}/trainer.py")
|
307 |
shutil.copy("script.py", f"{spacerunner_folder}/script.py")
|
308 |
shutil.copytree(dataset_folder, f"{spacerunner_folder}/{dataset_folder}")
|
309 |
+
requirements='''peft==0.7.1
|
310 |
-huggingface_hub
|
311 |
torch
|
312 |
git+https://github.com/huggingface/diffusers@5b186b712837104b40d095b26ed6a2ec61246cb4
|