multimodalart HF staff commited on
Commit
867b96b
1 Parent(s): e470146

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/main/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,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='''-peft
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