Update nodes.py

2025-02-26 11:59:05 +02:00
parent 1be6247e38
commit d138fccf80
1 changed files with 2 additions and 0 deletions
@@ -618,6 +618,7 @@ class WanVideoTextEncode:

        if force_offload:
            encoder.model.to(offload_device)
+            mm.soft_empty_cache()


        prompt_embeds_dict = {
@@ -667,6 +668,7 @@ class WanVideoImageClipEncode:
        clip_context = clip.visual(pixel_values)
        if force_offload:
            clip.model.to(offload_device)
+            mm.soft_empty_cache()

        aspect_ratio = H / W
        lat_h = round(