Spaces:

PAIR
/

StreamingT2V

Runtime error

hpoghos commited on Apr 9, 2024

Commit

870500e

•

1 Parent(s): 62db519

Update t2v_enhanced/model/video_ldm.py

Files changed (1) hide show

t2v_enhanced/model/video_ldm.py CHANGED Viewed

@@ -242,7 +242,7 @@ class VideoLDM(pl.LightningModule):
             else:
                 latents = None
             if idx == 0:
-                sample = cfg["video"]
             else:
                 if inference_params.conditioning_type == "fixed":
                     context = chunks_conditional[0][:self.unet_params.num_frame_conditioning]

             else:
                 latents = None
             if idx == 0:
+                sample = cfg["video"].to(self.device)
             else:
                 if inference_params.conditioning_type == "fixed":
                     context = chunks_conditional[0][:self.unet_params.num_frame_conditioning]