Spaces:
Running
Running
File size: 318 Bytes
708a6ea d4160b5 708a6ea d4160b5 8811d6d d4160b5 |
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("Lightricks/LTX-Video", torch_dtype=torch.bfloat16)
t2v_pipe.to(device)
i2v_pipe = LTXImageToVideoPipeline.from_pipe(t2v_pipe)
i2v_pipe.to(device) |