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 (
|