LeafCat79 commited on
Commit
9ad8fde
·
verified ·
1 Parent(s): 2d5e0e0

Update app.py

Browse files

Increase neural sprite detail

Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -124,9 +124,10 @@ HF_TOKEN = os.environ.get("HF_TOKEN", "")
124
  PRIMARY_IMAGE_MODEL = os.environ.get("PRIMARY_IMAGE_MODEL", "segmind/SSD-1B")
125
  PRIMARY_LORA_MODEL = os.environ.get("PRIMARY_LORA_MODEL", "latent-consistency/lcm-lora-ssd-1b")
126
  PRIMARY_IMAGE_STEPS = int(os.environ.get("PRIMARY_IMAGE_STEPS", "4"))
 
127
  PRIMARY_GUIDANCE_SCALE = float(os.environ.get("PRIMARY_GUIDANCE_SCALE", "1.5"))
128
  PRIMARY_BACKGROUND_STRENGTH = float(os.environ.get("PRIMARY_BACKGROUND_STRENGTH", "0.72"))
129
- PRIMARY_SPRITE_STRENGTH = float(os.environ.get("PRIMARY_SPRITE_STRENGTH", "0.72"))
130
  USE_PRIMARY_IMAGE_MODEL = os.environ.get("USE_PRIMARY_IMAGE_MODEL", "1") == "1"
131
  REQUIRE_PRIMARY_IMAGE_MODEL = os.environ.get(
132
  "REQUIRE_PRIMARY_IMAGE_MODEL",
@@ -1651,6 +1652,7 @@ def primary_diffusion_prompt(spec: AssetSpec) -> str:
1651
  return (
1652
  f"{spec.prompt} One complete isolated game sprite only, centered, fully visible, readable silhouette, "
1653
  "camera angle and pose exactly matching the supplied guide, on a plain uniform white studio background. "
 
1654
  "No scenery, floor, trees, foliage, frame, or environmental props."
1655
  )
1656
 
@@ -1685,7 +1687,7 @@ def primary_diffusion_png(spec: AssetSpec, index: int, run_id: int) -> tuple[byt
1685
  common = {
1686
  "prompt": primary_diffusion_prompt(spec),
1687
  "negative_prompt": diffusion_negative_prompt(spec),
1688
- "num_inference_steps": PRIMARY_IMAGE_STEPS,
1689
  "guidance_scale": PRIMARY_GUIDANCE_SCALE,
1690
  "generator": generator,
1691
  }
 
124
  PRIMARY_IMAGE_MODEL = os.environ.get("PRIMARY_IMAGE_MODEL", "segmind/SSD-1B")
125
  PRIMARY_LORA_MODEL = os.environ.get("PRIMARY_LORA_MODEL", "latent-consistency/lcm-lora-ssd-1b")
126
  PRIMARY_IMAGE_STEPS = int(os.environ.get("PRIMARY_IMAGE_STEPS", "4"))
127
+ PRIMARY_SPRITE_STEPS = int(os.environ.get("PRIMARY_SPRITE_STEPS", "6"))
128
  PRIMARY_GUIDANCE_SCALE = float(os.environ.get("PRIMARY_GUIDANCE_SCALE", "1.5"))
129
  PRIMARY_BACKGROUND_STRENGTH = float(os.environ.get("PRIMARY_BACKGROUND_STRENGTH", "0.72"))
130
+ PRIMARY_SPRITE_STRENGTH = float(os.environ.get("PRIMARY_SPRITE_STRENGTH", "0.82"))
131
  USE_PRIMARY_IMAGE_MODEL = os.environ.get("USE_PRIMARY_IMAGE_MODEL", "1") == "1"
132
  REQUIRE_PRIMARY_IMAGE_MODEL = os.environ.get(
133
  "REQUIRE_PRIMARY_IMAGE_MODEL",
 
1652
  return (
1653
  f"{spec.prompt} One complete isolated game sprite only, centered, fully visible, readable silhouette, "
1654
  "camera angle and pose exactly matching the supplied guide, on a plain uniform white studio background. "
1655
+ "Add recognizable costume, equipment, materials, and character details. "
1656
  "No scenery, floor, trees, foliage, frame, or environmental props."
1657
  )
1658
 
 
1687
  common = {
1688
  "prompt": primary_diffusion_prompt(spec),
1689
  "negative_prompt": diffusion_negative_prompt(spec),
1690
+ "num_inference_steps": PRIMARY_IMAGE_STEPS if is_background_spec(spec) else PRIMARY_SPRITE_STEPS,
1691
  "guidance_scale": PRIMARY_GUIDANCE_SCALE,
1692
  "generator": generator,
1693
  }