Upload modeling_videochat2_cls.py with huggingface_hub

modeling_videochat2_cls.py

@@ -17,6 +17,7 @@ class InternVideo2_cls(InternVideo2_VideoChat2):
                 self.model_config.llm.pretrained_llm_path,
                 torch_dtype=torch.bfloat16,
                 token=token,
+                num_labels=self.model_config.llm.num_labels
                 # attn_implementation="flash_attention_2",
             )
             self.lm = AutoModelForSequenceClassification.from_config(config)
@@ -88,6 +89,8 @@ class InternVideo2_cls(InternVideo2_VideoChat2):
         conversation += ("")
         return conversation
 
+    def test(self, x):
+        return x
 
 if __name__ == "__main__":