multimodalart HF staff commited on
Commit
21f25c7
1 Parent(s): c74a0ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,8 +46,8 @@ maximum_concepts = 3
46
  #Pre download the files
47
  if(is_gpu_associated):
48
  model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
49
- model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-1")
50
- model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-1-base")
51
  safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
52
  model_to_load = model_v1
53
 
 
46
  #Pre download the files
47
  if(is_gpu_associated):
48
  model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
49
+ model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-1", ignore_patterns=["*.ckpt", "*.safetensors"])
50
+ model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-1-base", ignore_patterns=["*.ckpt", "*.safetensors"])
51
  safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
52
  model_to_load = model_v1
53