From bf0bf2d5bafd7fffc46328521352c7fbfe71134e Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Tue, 4 Mar 2025 21:54:04 +0800 Subject: [PATCH] update wan examples --- examples/wanvideo/wan_14b_image_to_video.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/wanvideo/wan_14b_image_to_video.py b/examples/wanvideo/wan_14b_image_to_video.py index 4fe6be7..b429788 100644 --- a/examples/wanvideo/wan_14b_image_to_video.py +++ b/examples/wanvideo/wan_14b_image_to_video.py @@ -24,10 +24,10 @@ model_manager.load_models( "models/Wan-AI/Wan2.1-I2V-14B-480P/models_t5_umt5-xxl-enc-bf16.pth", "models/Wan-AI/Wan2.1-I2V-14B-480P/Wan2.1_VAE.pth", ], - torch_dtype=torch.float8_e4m3fn, # You can set `torch_dtype=torch.bfloat16` to disable FP8 quantization. + torch_dtype=torch.bfloat16, # You can set `torch_dtype=torch.float8_e4m3fn` to enable FP8 quantization. ) pipe = WanVideoPipeline.from_model_manager(model_manager, torch_dtype=torch.bfloat16, device="cuda") -pipe.enable_vram_management(num_persistent_param_in_dit=None) # You can set `num_persistent_param_in_dit` to a small number to reduce VRAM required. +pipe.enable_vram_management(num_persistent_param_in_dit=6*10**9) # You can set `num_persistent_param_in_dit` to a small number to reduce VRAM required. # Download example image dataset_snapshot_download(