Spaces:
Running on Zero
Running on Zero
ImageStudio Maintainer commited on
Commit ·
f4bbab1
1
Parent(s): f0b2701
feat(models): load wai-anima/samanima as Anime/Realistic, unify prefix
Browse files
app.py
CHANGED
|
@@ -195,7 +195,7 @@ IMAGE_MODELS = {
|
|
| 195 |
steps=9, guidance=0.0, height=1024, width=1024, negative=False, prefix="",
|
| 196 |
),
|
| 197 |
"wai-anima": dict(
|
| 198 |
-
label="
|
| 199 |
checkpoint="checkpoints/anima/waiANIMA_v10Base10.safetensors",
|
| 200 |
steps=24, guidance=0.0, height=1024, width=1024, negative=True,
|
| 201 |
prefix="masterpiece, best quality,score_7,",
|
|
@@ -204,10 +204,10 @@ IMAGE_MODELS = {
|
|
| 204 |
"blurry, jpeg artifacts, lowres,censor"),
|
| 205 |
),
|
| 206 |
"samanima": dict(
|
| 207 |
-
label="
|
| 208 |
checkpoint="checkpoints/anima/samANIMA_v20.safetensors",
|
| 209 |
steps=24, guidance=0.0, height=1024, width=1024, negative=True,
|
| 210 |
-
prefix="
|
| 211 |
default_negative=(
|
| 212 |
"missing finger,mutation,censor,censored,high contrast, harsh lighting, "
|
| 213 |
"cel shading, linear hatching,cartoon,toon,2D,CGI, comic, empty background, "
|
|
@@ -243,7 +243,7 @@ def _resolve_negative(entry, negative_prompt, use_negative_prompt, model_name):
|
|
| 243 |
|
| 244 |
# Which models to load this boot. Unknown keys are ignored; an empty / all-invalid
|
| 245 |
# selection falls back to the first registered model so the Space always boots.
|
| 246 |
-
DEFAULT_IMAGE_MODELS = "
|
| 247 |
_requested = [s.strip() for s in os.environ.get("IMAGE_MODELS", DEFAULT_IMAGE_MODELS).split(",") if s.strip()]
|
| 248 |
ENABLED_MODEL_KEYS = [k for k in _requested if k in IMAGE_MODELS] or [next(iter(IMAGE_MODELS))]
|
| 249 |
|
|
|
|
| 195 |
steps=9, guidance=0.0, height=1024, width=1024, negative=False, prefix="",
|
| 196 |
),
|
| 197 |
"wai-anima": dict(
|
| 198 |
+
label="Anime", family="anima",
|
| 199 |
checkpoint="checkpoints/anima/waiANIMA_v10Base10.safetensors",
|
| 200 |
steps=24, guidance=0.0, height=1024, width=1024, negative=True,
|
| 201 |
prefix="masterpiece, best quality,score_7,",
|
|
|
|
| 204 |
"blurry, jpeg artifacts, lowres,censor"),
|
| 205 |
),
|
| 206 |
"samanima": dict(
|
| 207 |
+
label="Realistic", family="anima",
|
| 208 |
checkpoint="checkpoints/anima/samANIMA_v20.safetensors",
|
| 209 |
steps=24, guidance=0.0, height=1024, width=1024, negative=True,
|
| 210 |
+
prefix="masterpiece, best quality, score_7,",
|
| 211 |
default_negative=(
|
| 212 |
"missing finger,mutation,censor,censored,high contrast, harsh lighting, "
|
| 213 |
"cel shading, linear hatching,cartoon,toon,2D,CGI, comic, empty background, "
|
|
|
|
| 243 |
|
| 244 |
# Which models to load this boot. Unknown keys are ignored; an empty / all-invalid
|
| 245 |
# selection falls back to the first registered model so the Space always boots.
|
| 246 |
+
DEFAULT_IMAGE_MODELS = "wai-anima,samanima"
|
| 247 |
_requested = [s.strip() for s in os.environ.get("IMAGE_MODELS", DEFAULT_IMAGE_MODELS).split(",") if s.strip()]
|
| 248 |
ENABLED_MODEL_KEYS = [k for k in _requested if k in IMAGE_MODELS] or [next(iter(IMAGE_MODELS))]
|
| 249 |
|