giers10/skymap-gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add prompt enrichment suffix for equirectangular 360 views

Browse Source
This commit is contained in:
Victor Giers
2026-05-07 16:18:57 +02:00
parent d846e0580d
commit e428c509d7
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
generate_equirect.py
View File

@@ -46,6 +46,7 @@ REALESRGAN_MODEL = "RealESRGAN_x4plus.pth"
REALESRGAN_SCALE = 4 # x4 model for full upscale REALESRGAN_SCALE = 4 # x4 model for full upscale
# Recommended VAE for SDXL checkpoints # Recommended VAE for SDXL checkpoints
SDXL_VAE = "stabilityai/sdxl-vae" SDXL_VAE = "stabilityai/sdxl-vae"
PROMPT_ENRICHMENT_SUFFIX = ", equirectangular 360 view"
def downloads_allowed() -> bool: def downloads_allowed() -> bool:
@@ -103,6 +104,13 @@ def sanitize_name(prompt: str) -> str:
return base or "env" return base or "env"
def enrich_prompt(prompt: str) -> str:
prompt = prompt.strip()
if prompt.endswith(PROMPT_ENRICHMENT_SUFFIX):
return prompt
return f"{prompt}{PROMPT_ENRICHMENT_SUFFIX}"
def next_filename(output_dir: str, base: str, width: int, height: int) -> str: def next_filename(output_dir: str, base: str, width: int, height: int) -> str:
os.makedirs(output_dir, exist_ok=True) os.makedirs(output_dir, exist_ok=True)
i = 1 i = 1