Hugging Face's logo

Spaces:
klemkeni
/
dreambooth-training
Build error

App Files Community
4
klemkeni commited on
Commit
456478a
•
1 Parent(s): d7dbf0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -31,8 +31,10 @@ maximum_concepts = 3
31
 
32
  #Pre download the files
33
  if(is_gpu_associated):
 
34
  model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
35
- model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
 
36
  model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
37
  safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
38
  model_to_load = model_v1
 
31
 
32
  #Pre download the files
33
  if(is_gpu_associated):
34
+
35
  model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
36
+ #model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
37
+ model_v2 = snapshot_download(repo_id="klemkeni/nicobooth1")
38
  model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
39
  safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
40
  model_to_load = model_v1