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

Update image saving logic to correctly handle prompts and save the final image

Browse Source
This commit is contained in:
Victor Giers
2026-05-07 11:05:05 +02:00
parent e32f6526f6
commit ad68d6bb72
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
generate_equirect.py
View File

@@ -12,7 +12,6 @@ import json
import math import math
import os import os
import re import re
import shutil
import tempfile import tempfile
import torch import torch
from PIL import Image, ImageDraw from PIL import Image, ImageDraw
@@ -555,7 +554,9 @@ def generate(
except Exception as e: # noqa: BLE001 except Exception as e: # noqa: BLE001
print(f"Upscaling failed ({upscale}); keeping seamless image: {e}") print(f"Upscaling failed ({upscale}); keeping seamless image: {e}")
shutil.move(final_path, output_path) with Image.open(final_path) as final_img:
final_img.load()
save_png_with_prompt(final_img, output_path, prompt)
try: try:
with Image.open(output_path) as _im: with Image.open(output_path) as _im:
print(f"→ Final image written to {output_path} [{_im.size[0]}x{_im.size[1]}]") print(f"→ Final image written to {output_path} [{_im.size[0]}x{_im.size[1]}]")