zhiweili commited on
Commit
8811d6d
·
1 Parent(s): 8a746ba

fix from_pipe

Browse files
Files changed (1) hide show
  1. 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.from_pipeline(t2v_pipe)
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)