Spaces:
Runtime error
Runtime error
feat: limit models amount
Browse files
app.py
CHANGED
@@ -25,14 +25,13 @@ import modules.safe as _
|
|
25 |
from modules.lora import LoRANetwork
|
26 |
|
27 |
models = [
|
28 |
-
|
29 |
-
("
|
30 |
-
("
|
31 |
-
("Pastal Mix", "andite/pastel-mix", 2),
|
32 |
-
("ACertainModel", "JosephusCheung/ACertainModel", 2),
|
33 |
]
|
34 |
|
35 |
-
base_name, base_model,
|
|
|
36 |
|
37 |
samplers_k_diffusion = [
|
38 |
("Euler a", "sample_euler_ancestral", {}),
|
@@ -132,8 +131,8 @@ def get_model(name):
|
|
132 |
return g_unet, g_lora
|
133 |
|
134 |
# precache on huggingface
|
135 |
-
|
136 |
-
|
137 |
|
138 |
def error_str(error, title="Error"):
|
139 |
return (
|
|
|
25 |
from modules.lora import LoRANetwork
|
26 |
|
27 |
models = [
|
28 |
+
("AbyssOrangeMix2", "Korakoe/AbyssOrangeMix2-HF"),
|
29 |
+
("Basil Mix", "nuigurumi/basil_mix"),
|
30 |
+
("Pastal Mix", "andite/pastel-mix"),
|
|
|
|
|
31 |
]
|
32 |
|
33 |
+
base_name, base_model, = models[0]
|
34 |
+
clip_skip = 2
|
35 |
|
36 |
samplers_k_diffusion = [
|
37 |
("Euler a", "sample_euler_ancestral", {}),
|
|
|
131 |
return g_unet, g_lora
|
132 |
|
133 |
# precache on huggingface
|
134 |
+
for model in get_model_list():
|
135 |
+
get_model(model[0])
|
136 |
|
137 |
def error_str(error, title="Error"):
|
138 |
return (
|