From e428c509d7f9d63a935e1ca46b82f163ea60b643 Mon Sep 17 00:00:00 2001 From: Victor Giers Date: Thu, 7 May 2026 16:18:57 +0200 Subject: [PATCH] Add prompt enrichment suffix for equirectangular 360 views --- generate_equirect.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/generate_equirect.py b/generate_equirect.py index 571eb5d..46020b4 100644 --- a/generate_equirect.py +++ b/generate_equirect.py @@ -46,6 +46,7 @@ REALESRGAN_MODEL = "RealESRGAN_x4plus.pth" REALESRGAN_SCALE = 4 # x4 model for full upscale # Recommended VAE for SDXL checkpoints SDXL_VAE = "stabilityai/sdxl-vae" +PROMPT_ENRICHMENT_SUFFIX = ", equirectangular 360 view" def downloads_allowed() -> bool: @@ -103,6 +104,13 @@ def sanitize_name(prompt: str) -> str: 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: os.makedirs(output_dir, exist_ok=True) i = 1