Reduce latent upscale branch memory
This commit is contained in:
@@ -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"
|
||||||
|
|||||||
Reference in New Issue
Block a user