AI/ComfyUI-WanVideoWrapper
1
0
Fork 0
mirror of https://github.com/kijai/ComfyUI-WanVideoWrapper.git synced 2026-01-26 23:41:35 +03:00
Code Activity

free VAE cache

Browse Source 
Somehow missed this, should free up 1-2GB VRAM
This commit is contained in:
kijai
2025-02-26 15:54:59 +02:00
parent d5721e3ef8
commit be2f557fa6
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
nodes.py
View File

@@ -1019,6 +1019,8 @@ class WanVideoDecode:
print(image.shape)
print(image.min(), image.max())
vae.to(offload_device)
vae.model.clear_cache()
mm.soft_empty_cache()
image = (image - image.min()) / (image.max() - image.min())
image = torch.clamp(image, 0.0, 1.0)
@@ -1061,12 +1063,13 @@ class WanVideoEncode:
if noise_aug_strength > 0.0:
image = add_noise_to_reference_video(image, ratio=noise_aug_strength)
latents = vae.encode(image, device=device, tiled=enable_vae_tiling, tile_size=(tile_x, tile_y), tile_stride=(tile_stride_x, tile_stride_y))#.latent_dist.sample(generator)
latents = vae.encode(image, device=device, tiled=enable_vae_tiling, tile_size=(tile_x, tile_y), tile_stride=(tile_stride_x, tile_stride_y))
if latent_strength != 1.0:
latents *= latent_strength
#latents = latents * vae.config.scaling_factor
vae.to(offload_device)
vae.model.clear_cache()
mm.soft_empty_cache()
print("encoded latents shape",latents.shape)