Commit
•
21f25c7
1
Parent(s):
c74a0ea
Update app.py
Browse files
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 |
|