Reduce latent upscale branch memory

This commit is contained in:
2026-09-03 18:08:44 +00:00
parent f7b94ccaef
commit 51e03a39b7
+2
View File
@@ -6818,6 +6818,8 @@ class H3LongVideos:
refined_out = {"samples": comfy.nested_tensor.NestedTensor((refined_video, full_audio))} refined_out = {"samples": comfy.nested_tensor.NestedTensor((refined_video, full_audio))}
timing["latent_upscale_sample"] += time.perf_counter() - latent_start timing["latent_upscale_sample"] += time.perf_counter() - latent_start
refined_out = _video_only_refined_latent(upscale_latent, refined_out) refined_out = _video_only_refined_latent(upscale_latent, refined_out)
del upscale_latent, upscaled_video, full_audio
mm.soft_empty_cache()
except Exception as e: except Exception as e:
if _is_oom(e): if _is_oom(e):
e._h3_stage = "sampling" e._h3_stage = "sampling"