File size: 329 Bytes
ab9e6f4 |
1 2 3 4 5 6 7 8 9 10 11 12 |
import torch
from diffusers import LTXPipeline, LTXImageToVideoPipeline
device = "cuda" if torch.cuda.is_available() else "cpu"
t2v_pipe = LTXPipeline.from_pretrained("Skywork/SkyReels-V1-Hunyuan-T2V", torch_dtype=torch.bfloat16)
t2v_pipe.to(device)
i2v_pipe = LTXImageToVideoPipeline.from_pipe(t2v_pipe)
i2v_pipe.to(device) |