VideoLLaMA2

Runtime error

ClownRat commited on Oct 17, 2024

Commit

d078991

1 Parent(s): 1d5815e

Fix typos.

Files changed (1) hide show

VideoLLaMA2/videollama2/model/videollama2_arch.py CHANGED Viewed

@@ -117,7 +117,7 @@ class Videollama2MetaForCausalLM(ABC):
         data_batch = []
         for i, (data, modal) in enumerate(images):
-            print(data, modal.shape)
             if modal == 'image':
                 data = data.expand(num_frames, -1, -1, -1)
             else:

         data_batch = []
         for i, (data, modal) in enumerate(images):
+            print(data.shape, modal)
             if modal == 'image':
                 data = data.expand(num_frames, -1, -1, -1)
             else: