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

auto-git:

Browse Source 
[change] generate_equirect.py
This commit is contained in:
Victor Giers
2026-05-07 12:11:33 +02:00
parent a5d9e225ba
commit bfbe76b700
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
generate_equirect.py
View File

@@ -122,10 +122,12 @@ def clear_torch_cache(device: str | None = None) -> None:
empty_cache() empty_cache()
def configure_pipeline_memory(pipe) -> None: def configure_pipeline_memory(pipe, *, vae_tiling: bool = True) -> None:
pipe.enable_attention_slicing() pipe.enable_attention_slicing()
if hasattr(pipe, "enable_vae_tiling"): if vae_tiling and hasattr(pipe, "enable_vae_tiling"):
pipe.enable_vae_tiling() pipe.enable_vae_tiling()
elif not vae_tiling and hasattr(pipe, "disable_vae_tiling"):
pipe.disable_vae_tiling()
if hasattr(pipe, "enable_vae_slicing"): if hasattr(pipe, "enable_vae_slicing"):
pipe.enable_vae_slicing() pipe.enable_vae_slicing()
@@ -357,7 +359,7 @@ def postprocess_image(
safety_checker=None, safety_checker=None,
requires_safety_checker=False, requires_safety_checker=False,
).to(device) ).to(device)
configure_pipeline_memory(inpaint_pipe) configure_pipeline_memory(inpaint_pipe, vae_tiling=False)
print("→ Inpainting seam for seamless tiling…") print("→ Inpainting seam for seamless tiling…")
progress_cb("inpaint", 0, steps) progress_cb("inpaint", 0, steps)