Update README.md
Browse files
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("
|
| 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("
|
| 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
|