jingulee commited on
Commit
15f1944
·
verified ·
1 Parent(s): 640e8c2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -61,7 +61,7 @@ pip install -U diffusers
61
  from diffusers import WanPipeline, UniPCMultistepScheduler
62
 
63
  device = "cuda"
64
- pipe = WanPipeline.from_pretrained("jingulee/Wan2.2-TI2V-5B-Turbo-Diffusers", torch_dtype=torch.bfloat16).to(device)
65
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
66
 
67
  width = 1280
@@ -93,7 +93,7 @@ from diffusers import UniPCMultistepScheduler, WanImageToVideoPipeline
93
  from diffusers.utils import export_to_video, load_image
94
 
95
  device = "cuda"
96
- pipe = WanImageToVideoPipeline.from_pretrained("jingulee/Wan2.2-TI2V-5B-Turbo-Diffusers", torch_dtype=torch.bfloat16).to(device)
97
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
98
 
99
  max_area = 1280 * 704
 
61
  from diffusers import WanPipeline, UniPCMultistepScheduler
62
 
63
  device = "cuda"
64
+ pipe = WanPipeline.from_pretrained("yetter-ai/Wan2.2-TI2V-5B-Turbo-Diffusers", torch_dtype=torch.bfloat16).to(device)
65
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
66
 
67
  width = 1280
 
93
  from diffusers.utils import export_to_video, load_image
94
 
95
  device = "cuda"
96
+ pipe = WanImageToVideoPipeline.from_pretrained("yetter-ai/Wan2.2-TI2V-5B-Turbo-Diffusers", torch_dtype=torch.bfloat16).to(device)
97
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
98
 
99
  max_area = 1280 * 704