Spaces:
Running
on
T4
Running
on
T4
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] = {
|