support wanxvae

examples/WanX/test_vae.py

@@ -0,0 +1,18 @@
+import torch
+from diffsynth.models.wanx_vae import WanXVideoVAE
+from diffsynth import ModelManager
+
+
+model_manager = ModelManager(torch_dtype=torch.float, device="cuda")
+model_manager.load_models([
+    "models/WanX/vae.pth",
+])
+
+vae = model_manager.fetch_model('wanxvideo_vae')
+
+latents = [torch.load('sample.pt')]
+videos = vae.decode(latents)
+back_encode = vae.encode(videos)
+print(latents)
+print(videos)
+print(back_encode)