Upload InternVideo2_Classification_test

config.json

@@ -1,7 +1,13 @@
 {
+  "_name_or_path": "OpenGVLab/InternVideo2-Chat-8B",
+  "architectures": [
+    "InternVideo2_Classification_test"
+  ],
   "auto_map": {
-    "AutoConfig": "model_config.VideoChat2Config"
+    "AutoConfig": "OpenGVLab/InternVideo2-Chat-8B--model_config.VideoChat2Config",
+    "AutoModel": "modeling_videochat2_classification.InternVideo2_Classification_test"
   },
+  "model_cls": "InternVideo2_VideoChat2",
   "model_config": {
     "bridge": {
       "extra_num_query_token": 64,

modeling_videochat2_classification.py

@@ -384,6 +384,19 @@ class InternVideo2_Classification(PreTrainedModel):
     def device(self):
         return self.lm.device
 
+
+class InternVideo2_Classification_test(PreTrainedModel):
+    config_class = VideoChat2Config
+    def __init__(self, config):
+        self.model_config = config.model_config
+        # config.model_config = None
+        super().__init__(config)
+        self.w = torch.randn(10,10, requires_grad=True)
+
+
+    def forward(self, x):
+        return x
+
 if __name__ == "__main__":
 
     tokenizer =  AutoTokenizer.from_pretrained('OpenGVLab/InternVideo2-Chat-8B',trust_remote_code=True,use_fast=False)