| 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) |