multimodalart HF staff commited on
Commit
08be887
1 Parent(s): 3085a9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -304,6 +304,7 @@ def start_training(
304
  shutil.copy("train_dreambooth_lora_sdxl_advanced.py", f"{spacerunner_folder}/script.py")
305
  shutil.copytree(dataset_folder, f"{spacerunner_folder}/{dataset_folder}")
306
  requirements='''-peft
 
307
  torch
308
  git+https://github.com/huggingface/diffusers@c05d71be04345b18a5120542c363f6e4a3f99b05
309
  transformers
@@ -311,7 +312,8 @@ accelerate
311
  safetensors
312
  prodigyopt
313
  hf-transfer
314
- git+https://github.com/huggingface/datasets.git'''
 
315
  file_path = f'{spacerunner_folder}/requirements.txt'
316
  with open(file_path, 'w') as file:
317
  file.write(requirements)
 
304
  shutil.copy("train_dreambooth_lora_sdxl_advanced.py", f"{spacerunner_folder}/script.py")
305
  shutil.copytree(dataset_folder, f"{spacerunner_folder}/{dataset_folder}")
306
  requirements='''-peft
307
+ -huggingface_hub
308
  torch
309
  git+https://github.com/huggingface/diffusers@c05d71be04345b18a5120542c363f6e4a3f99b05
310
  transformers
 
312
  safetensors
313
  prodigyopt
314
  hf-transfer
315
+ git+https://github.com/huggingface/datasets.git
316
+ git+https://github.com/huggingface/huggingface_hub.git@8d052492fe0059c606c1a48d7a914b15b64a834d'''
317
  file_path = f'{spacerunner_folder}/requirements.txt'
318
  with open(file_path, 'w') as file:
319
  file.write(requirements)