FastDM
/

Wan2.2-T2V-A14B-Merge-Lightning-V1.0-Diffusers

Model card Files Files and versions

Rayzl commited on Sep 12

Commit

4ad7468

·

verified ·

1 Parent(s): d42441b

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -16,8 +16,8 @@ from diffusers.utils import export_to_video, load_image
 dtype = torch.bfloat16
 device = "cuda:2"
-vae = AutoencoderKLWan.from_pretrained("FastDM/Wan2.2-T2V-A14B-Merge-Lightning-V1.1-Diffusers", subfolder="vae", torch_dtype=torch.float32)
-pipe = WanPipeline.from_pretrained("FastDM/Wan2.2-T2V-A14B-Merge-Lightning-V1.1-Diffusers", vae=vae, torch_dtype=dtype)
 pipe.to(device)
 height = 720
@@ -42,5 +42,5 @@ export_to_video(output, "t2v_out.mp4", fps=16)
 test command:
 ```
-python gen.py --model-path FastDM/Wan2.2-T2V-A14B-Merge-Lightning-V1.1-Diffusers --architecture wan --guidance-scale 1.0 --height 720 --width 1280 --steps 4 --use-fp8 --output-path ./wan-a14b-lightningv1.1-fp8-guid1.mp4 --num-frames 81 --fps 16 --prompts "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."
 ```

 dtype = torch.bfloat16
 device = "cuda:2"
+vae = AutoencoderKLWan.from_pretrained("FastDM/Wan2.2-T2V-A14B-Merge-Lightning-V1.0-Diffusers", subfolder="vae", torch_dtype=torch.float32)
+pipe = WanPipeline.from_pretrained("FastDM/Wan2.2-T2V-A14B-Merge-Lightning-V1.0-Diffusers", vae=vae, torch_dtype=dtype)
 pipe.to(device)
 height = 720
 test command:
 ```
+python gen.py --model-path FastDM/Wan2.2-T2V-A14B-Merge-Lightning-V1.0-Diffusers --architecture wan --guidance-scale 1.0 --height 720 --width 1280 --steps 4 --use-fp8 --output-path ./wan-a14b-lightningv1.1-fp8-guid1.mp4 --num-frames 81 --fps 16 --prompts "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."
 ```