linoyts HF Staff commited on
Commit
21c3a50
·
verified ·
1 Parent(s): 20d0bc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -283,6 +283,8 @@ class AudioConditionedICLoraPipeline(ICLoraPipeline):
283
  height=stage_2_output_shape.height,
284
  width=stage_2_output_shape.width,
285
  video_encoder=video_encoder,
 
 
286
  # num_frames=num_frames,
287
  # conditioning_attention_strength=conditioning_attention_strength,
288
  # conditioning_attention_mask=conditioning_attention_mask,
 
283
  height=stage_2_output_shape.height,
284
  width=stage_2_output_shape.width,
285
  video_encoder=video_encoder,
286
+ dtype=dtype,
287
+ device=self.device,
288
  # num_frames=num_frames,
289
  # conditioning_attention_strength=conditioning_attention_strength,
290
  # conditioning_attention_mask=conditioning_attention_mask,