ClownRat commited on
Commit
4f7d7a4
•
1 Parent(s): d078991

Clean debug strings.

Browse files
VideoLLaMA2/videollama2/model/videollama2_arch.py CHANGED
@@ -117,7 +117,6 @@ class Videollama2MetaForCausalLM(ABC):
117
 
118
  data_batch = []
119
  for i, (data, modal) in enumerate(images):
120
- print(data.shape, modal)
121
  if modal == 'image':
122
  data = data.expand(num_frames, -1, -1, -1)
123
  else:
@@ -126,8 +125,6 @@ class Videollama2MetaForCausalLM(ABC):
126
 
127
  data_batch = torch.stack(data_batch, dim=0)
128
 
129
- print(data_batch.shape)
130
-
131
  assert len(data_batch.size()) == 5
132
  batch_size = data_batch.size(0)
133
 
 
117
 
118
  data_batch = []
119
  for i, (data, modal) in enumerate(images):
 
120
  if modal == 'image':
121
  data = data.expand(num_frames, -1, -1, -1)
122
  else:
 
125
 
126
  data_batch = torch.stack(data_batch, dim=0)
127
 
 
 
128
  assert len(data_batch.size()) == 5
129
  batch_size = data_batch.size(0)
130