morpheushoc commited on
Commit
f408a17
·
verified ·
1 Parent(s): 64ed454

Upload config

Browse files
Files changed (2) hide show
  1. config.json +3 -0
  2. model_config.py +12 -6
config.json CHANGED
@@ -1,4 +1,7 @@
1
  {
 
 
 
2
  "model_config": {
3
  "bridge": {
4
  "extra_num_query_token": 64,
 
1
  {
2
+ "auto_map": {
3
+ "AutoConfig": "model_config.VideoChat2Config"
4
+ },
5
  "model_config": {
6
  "bridge": {
7
  "extra_num_query_token": 64,
model_config.py CHANGED
@@ -2,12 +2,18 @@ from transformers.configuration_utils import PretrainedConfig
2
  from easydict import EasyDict as MyEasyDict
3
 
4
 
 
 
 
 
 
 
 
 
 
 
5
  class VideoChat2Config(PretrainedConfig):
6
  model_type = 'InternVideo2_VideoChat2'
7
 
8
- def __init__(
9
- self,
10
- model_config=None,
11
- **kwargs):
12
- super().__init__(**kwargs)
13
- self.model_config = MyEasyDict(model_config)
 
2
  from easydict import EasyDict as MyEasyDict
3
 
4
 
5
+ # class VideoChat2Config(PretrainedConfig):
6
+ # model_type = 'InternVideo2_VideoChat2'
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_VideoChat2'
17
 
18
+ def __init__(self,**kwargs):
19
+ super().__init__(**kwargs)