video_model.py

import torch

from diffusers import LTXPipeline, LTXImageToVideoPipeline


device = "cuda" if torch.cuda.is_available() else "cpu"

t2v_pipe = LTXPipeline.from_pretrained("Lightricks/LTX-Video", torch_dtype=torch.bfloat16)
t2v_pipe.to(device)

i2v_pipe = LTXImageToVideoPipeline.from_pretrained("Lightricks/LTX-Image", torch_dtype=torch.bfloat16)
i2v_pipe.to(device)