sayakpaul HF Staff commited on
Commit
6a0340f
·
1 Parent(s): 5460f04
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -32,9 +32,12 @@ MIN_DURATION = round(MIN_FRAMES_MODEL / FIXED_FPS, 1)
32
  MAX_DURATION = round(MAX_FRAMES_MODEL / FIXED_FPS, 1)
33
 
34
  pipe = LTXConditionPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16).to("cuda")
 
 
 
35
  optimize_pipeline_(
36
  pipe,
37
- image=Image.new("RGB", (LANDSCAPE_WIDTH, LANDSCAPE_HEIGHT)),
38
  prompt="prompt",
39
  height=LANDSCAPE_HEIGHT,
40
  width=LANDSCAPE_WIDTH,
 
32
  MAX_DURATION = round(MAX_FRAMES_MODEL / FIXED_FPS, 1)
33
 
34
  pipe = LTXConditionPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16).to("cuda")
35
+ dummy_image = Image.new("RGB", (LANDSCAPE_WIDTH, LANDSCAPE_HEIGHT))
36
+ video = load_video(export_to_video([dummy_image]))
37
+ dummy_cond = LTXVideoCondition(video=video, frame_index=0)
38
  optimize_pipeline_(
39
  pipe,
40
+ conditions=[dummy_cond],
41
  prompt="prompt",
42
  height=LANDSCAPE_HEIGHT,
43
  width=LANDSCAPE_WIDTH,