Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
·
8811d6d
1
Parent(s):
8a746ba
fix from_pipe
Browse files- video_model.py +1 -1
video_model.py
CHANGED
@@ -8,5 +8,5 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
8 |
t2v_pipe = LTXPipeline.from_pretrained("Lightricks/LTX-Video", torch_dtype=torch.bfloat16)
|
9 |
t2v_pipe.to(device)
|
10 |
|
11 |
-
i2v_pipe = LTXImageToVideoPipeline.
|
12 |
i2v_pipe.to(device)
|
|
|
8 |
t2v_pipe = LTXPipeline.from_pretrained("Lightricks/LTX-Video", torch_dtype=torch.bfloat16)
|
9 |
t2v_pipe.to(device)
|
10 |
|
11 |
+
i2v_pipe = LTXImageToVideoPipeline.from_pipe(t2v_pipe)
|
12 |
i2v_pipe.to(device)
|