auto-git:

[change] generate_equirect.py
This commit is contained in:
2026-05-07 12:02:19 +02:00
parent 470dc0df12
commit 7758a114f7

View File

@@ -122,6 +122,28 @@ def clear_torch_cache(device: str | None = None) -> None:
empty_cache()
def configure_pipeline_memory(pipe) -> None:
pipe.enable_attention_slicing()
if hasattr(pipe, "enable_vae_tiling"):
pipe.enable_vae_tiling()
if hasattr(pipe, "enable_vae_slicing"):
pipe.enable_vae_slicing()
def make_progress_cb(enable_upscale: bool, seam_inpaint: bool):
def progress_cb(phase: str, current: int, total: int):
payload = {
"phase": phase,
"current": current,
"total": total,
"upscale": enable_upscale,
"seamInpaint": seam_inpaint,
}
print(f"PROGRESS {json.dumps(payload)}", flush=True)
return progress_cb
def decode_latents_to_image(vae: AutoencoderKL, latents: torch.Tensor, device: str) -> Image.Image:
print("→ Decoding latent image with standalone VAE…")
if hasattr(vae, "enable_tiling"):