Upload config

config.json

@@ -3,6 +3,7 @@
     "InternVideo2_Classification"
   ],
   "auto_map": {
+    "AutoConfig": "model_config.VideoChat2Config",
     "AutoModel": "modeling_videochat2_classification.InternVideo2_Classification"
   },
   "model_config": {

model_config.py

@@ -2,18 +2,18 @@ from transformers.configuration_utils import PretrainedConfig
 from easydict import EasyDict as MyEasyDict
 
 
-# class VideoChat2Config(PretrainedConfig):
-#     model_type = 'InternVideo2_VideoChat2'
+class VideoChat2Config(PretrainedConfig):
+    model_type = 'InternVideo2_Classification_test'
 
-#     def __init__(
-#             self,
-#             model_config=None,
-#             **kwargs):
-#         super().__init__(**kwargs)
-#         self.model_config  = MyEasyDict(model_config)
+    def __init__(
+            self,
+            model_config=None,
+            **kwargs):
+        super().__init__(**kwargs)
+        self.model_config  = MyEasyDict(model_config)
 
-class VideoChat2Config(PretrainedConfig):
-    model_type = 'InternVideo2_VideoChat2'
+# class VideoChat2Config(PretrainedConfig):
+#     model_type = 'InternVideo2_Classification_test'
 
-    def __init__(self,**kwargs):
-        super().__init__(**kwargs)
+#     def __init__(self,**kwargs):
+#         super().__init__(**kwargs)