ImageStudio Maintainer commited on
Commit
f4bbab1
·
1 Parent(s): f0b2701

feat(models): load wai-anima/samanima as Anime/Realistic, unify prefix

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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="WAI-ANIMA", 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,10 +204,10 @@ IMAGE_MODELS = {
204
  "blurry, jpeg artifacts, lowres,censor"),
205
  ),
206
  "samanima": dict(
207
- label="SamANIMA", family="anima",
208
  checkpoint="checkpoints/anima/samANIMA_v20.safetensors",
209
  steps=24, guidance=0.0, height=1024, width=1024, negative=True,
210
- prefix="score_9,score_6,@2025,@candid photo,newest,HDR, photography, (dramatic lighting), very aesthetic",
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 = "one-obsession,moody-pro-mix"
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