Spaces:
Configuration error
Configuration error
Update app.py
Browse files
app.py
CHANGED
@@ -34,12 +34,12 @@ 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 |
-
|
38 |
-
|
39 |
-
|
40 |
-
model_to_load =
|
41 |
-
|
42 |
-
|
43 |
|
44 |
def swap_text(option, base):
|
45 |
resize_width = 768 if base == "v2-768" else 512
|
|
|
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 |
+
model_alt = snapshot_download(repo_id="BAAI/AltDiffusion")
|
38 |
+
model_alt_m9 = snapshot_download(repo_id="BAAI/AltDiffusion-m9")
|
39 |
+
safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
|
40 |
+
model_to_load = model_alt_m9
|
41 |
+
with zipfile.ZipFile("mix.zip", 'r') as zip_ref:
|
42 |
+
zip_ref.extractall(".")
|
43 |
|
44 |
def swap_text(option, base):
|
45 |
resize_width = 768 if base == "v2-768" else 512
|