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
|