morpheushoc commited on
Commit
819249b
·
verified ·
1 Parent(s): a9e49d0

Upload InternVideo2_Classification_test

Browse files
Files changed (3) hide show
  1. config.json +1 -0
  2. model.safetensors +1 -1
  3. model_config.py +19 -0
config.json CHANGED
@@ -3,6 +3,7 @@
3
  "InternVideo2_Classification_test"
4
  ],
5
  "auto_map": {
 
6
  "AutoModel": "modeling_videochat2_classification.InternVideo2_Classification_test"
7
  },
8
  "model_config": {
 
3
  "InternVideo2_Classification_test"
4
  ],
5
  "auto_map": {
6
+ "AutoConfig": "model_config.VideoChat2Config",
7
  "AutoModel": "modeling_videochat2_classification.InternVideo2_Classification_test"
8
  },
9
  "model_config": {
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dad2c8637e76385541187f8578c922050cc7b9c704e774f1fdab50b623f3b517
3
  size 743024240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5164127ec718b2fb3abdc54e12a4dfaae390762fb1936a96dfec9c61485e6d39
3
  size 743024240
model_config.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers.configuration_utils import PretrainedConfig
2
+ from easydict import EasyDict as MyEasyDict
3
+
4
+
5
+ class VideoChat2Config(PretrainedConfig):
6
+ model_type = 'InternVideo2_Classification_test'
7
+
8
+ def __init__(
9
+ self,
10
+ model_config=None,
11
+ **kwargs):
12
+ super().__init__(**kwargs)
13
+ self.model_config = MyEasyDict(model_config)
14
+
15
+ # class VideoChat2Config(PretrainedConfig):
16
+ # model_type = 'InternVideo2_Classification_test'
17
+
18
+ # def __init__(self,**kwargs):
19
+ # super().__init__(**kwargs)