lora-ease-dk

Runtime error

dkebudi commited on Jul 30, 2024

Commit

e055b3b

verified ·

1 Parent(s): e374933

switch out the training script to custom training

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ MAX_IMAGES = 500
 is_spaces = True if os.environ.get('SPACE_ID') else False
-training_script_url = "https://raw.githubusercontent.com/huggingface/diffusers/ba28006f8b2a0f7ec3b6784695790422b4f80a97/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py"
 subprocess.run(['wget', '-N', training_script_url])
 orchestrator_script_url = "https://huggingface.co/datasets/multimodalart/lora-ease-helper/raw/main/script.py"
 subprocess.run(['wget', '-N', orchestrator_script_url])

 is_spaces = True if os.environ.get('SPACE_ID') else False
+training_script_url = "https://huggingface.co/spaces/dkebudi/lora-ease-dk/raw/main/train_dreambooth_lora_sdxl_advanced.py"
 subprocess.run(['wget', '-N', training_script_url])
 orchestrator_script_url = "https://huggingface.co/datasets/multimodalart/lora-ease-helper/raw/main/script.py"
 subprocess.run(['wget', '-N', orchestrator_script_url])