offload before vae encode too in WanAnimate loop

2025-09-20 21:52:37 +03:00
parent b9fd93c19e
commit a6dc8aeb64
1 changed files with 3 additions and 0 deletions
@@ -4324,6 +4324,9 @@ class WanVideoSampler:
                            self.cache_state = [None, None]
                          
                            if ref_latent is not None:
+                                if offload:
+                                    offload_transformer(transformer)
+                                    offloaded = True
                                vae.to(device)
                                if ref_masks is not None:
                                    msk = ref_masks_in[:, start_latent:end_latent].to(device, dtype)