import torch

from diffusers import LTXPipeline


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

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