Hugging Face's logo

Spaces:
marcus2000
/
dreambooth-training
Runtime error

App Files Community
multimodalart HF staff commited on
Commit
2e29bd9
•
1 Parent(s): 173552f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ maximum_concepts = 3
31
  #Pre download the files
32
  model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
33
  #model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
34
- model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
35
  safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
36
 
37
  model_to_load = model_v1
 
31
  #Pre download the files
32
  model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
33
  #model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
34
+ model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base", revision="fp16")
35
  safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
36
 
37
  model_to_load = model_v1