Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ from diffusers import (
|
|
| 11 |
DDIMScheduler,
|
| 12 |
UNet2DConditionModel,
|
| 13 |
)
|
| 14 |
-
from modules.
|
| 15 |
CrossAttnProcessor,
|
| 16 |
StableDiffusionPipeline,
|
| 17 |
load_lora_attn_procs,
|
|
@@ -28,6 +28,7 @@ models = [
|
|
| 28 |
("Anything 4.0", "andite/anything-v4.0"),
|
| 29 |
("Open Journey", "prompthero/openjourney"),
|
| 30 |
("Basil Mix", "nuigurumi/basil_mix"),
|
|
|
|
| 31 |
("ACertainModel", "JosephusCheung/ACertainModel"),
|
| 32 |
]
|
| 33 |
|
|
@@ -98,11 +99,7 @@ if torch.cuda.is_available():
|
|
| 98 |
|
| 99 |
|
| 100 |
def get_model_list():
|
| 101 |
-
|
| 102 |
-
for model in models:
|
| 103 |
-
if Path(model[1]).is_dir():
|
| 104 |
-
model_available.append(model)
|
| 105 |
-
return model_available
|
| 106 |
|
| 107 |
unet_cache = {
|
| 108 |
base_name: unet
|
|
@@ -297,7 +294,7 @@ def detect_text(text, state, width, height):
|
|
| 297 |
new_state[item] = {
|
| 298 |
"map": state[item]["map"],
|
| 299 |
"weight": state[item]["weight"],
|
| 300 |
-
"mask_outsides": state[item]["
|
| 301 |
}
|
| 302 |
else:
|
| 303 |
new_state[item] = {
|
|
|
|
| 11 |
DDIMScheduler,
|
| 12 |
UNet2DConditionModel,
|
| 13 |
)
|
| 14 |
+
from modules.model import (
|
| 15 |
CrossAttnProcessor,
|
| 16 |
StableDiffusionPipeline,
|
| 17 |
load_lora_attn_procs,
|
|
|
|
| 28 |
("Anything 4.0", "andite/anything-v4.0"),
|
| 29 |
("Open Journey", "prompthero/openjourney"),
|
| 30 |
("Basil Mix", "nuigurumi/basil_mix"),
|
| 31 |
+
("Pastal Mix", "andite/pastel-mix")
|
| 32 |
("ACertainModel", "JosephusCheung/ACertainModel"),
|
| 33 |
]
|
| 34 |
|
|
|
|
| 99 |
|
| 100 |
|
| 101 |
def get_model_list():
|
| 102 |
+
return models
|
|
|
|
|
|
|
|
|
|
|
|
|
| 103 |
|
| 104 |
unet_cache = {
|
| 105 |
base_name: unet
|
|
|
|
| 294 |
new_state[item] = {
|
| 295 |
"map": state[item]["map"],
|
| 296 |
"weight": state[item]["weight"],
|
| 297 |
+
"mask_outsides": state[item]["mask_outsides"],
|
| 298 |
}
|
| 299 |
else:
|
| 300 |
new_state[item] = {
|