starsfriday commited on
Commit
0b7032e
·
verified ·
1 Parent(s): fbdc517

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -83,13 +83,13 @@ from diffusers import AutoencoderKLWan, WanImageToVideoPipeline
83
  from transformers import CLIPVisionModel
84
  import numpy as np
85
 
86
- model_id = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
87
  image_encoder = CLIPVisionModel.from_pretrained(model_id, subfolder="image_encoder", torch_dtype=torch.float32)
88
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
89
  pipe = WanImageToVideoPipeline.from_pretrained(model_id, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16)
90
  pipe.to("cuda")
91
 
92
- pipe.load_lora_weights("Remade/Squish")
93
 
94
  pipe.enable_model_cpu_offload() #for low-vram environments
95
 
 
83
  from transformers import CLIPVisionModel
84
  import numpy as np
85
 
86
+ model_id = "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers"
87
  image_encoder = CLIPVisionModel.from_pretrained(model_id, subfolder="image_encoder", torch_dtype=torch.float32)
88
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
89
  pipe = WanImageToVideoPipeline.from_pretrained(model_id, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16)
90
  pipe.to("cuda")
91
 
92
+ pipe.load_lora_weights("valiantcat/Wan2.1-Fight-LoRA")
93
 
94
  pipe.enable_model_cpu_offload() #for low-vram environments
95