Spaces:
Running on Zero
Running on Zero
Update app.py
Browse filesIncrease neural sprite detail
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.
|
| 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 |
}
|