diff --git a/generate_equirect.py b/generate_equirect.py index 46020b4..8c60a36 100644 --- a/generate_equirect.py +++ b/generate_equirect.py @@ -545,6 +545,7 @@ def generate( height: int = 512, seam_inpaint: bool = False, ) -> str: + prompt = enrich_prompt(prompt) model_path = resolve_generation_model_path(model_path, work_dir) device = select_device() @@ -868,13 +869,14 @@ def main(): print(f"Generation failed: {e}") raise - base = sanitize_name(args.prompt) + prompt = enrich_prompt(args.prompt) + base = sanitize_name(prompt) target = args.output or next_filename(args.output_dir, base, args.width, args.height) output_abs = os.path.abspath(target) try: result_path = generate( - args.prompt, + prompt, output_abs, args.work_dir, upscale=args.upscale,