File size: 548 Bytes
95f05fb b6fd19a 5d4a847 f408a17 b6fd19a f408a17 b6fd19a 95f05fb b6fd19a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from transformers.configuration_utils import PretrainedConfig
from easydict import EasyDict as MyEasyDict
class VideoChat2Config(PretrainedConfig):
model_type = 'InternVideo2_Classification'
def __init__(
self,
model_config=None,
**kwargs):
super().__init__(**kwargs)
self.model_config = MyEasyDict(model_config)
# class VideoChat2Config(PretrainedConfig):
# model_type = 'InternVideo2_Classification_test'
# def __init__(self,**kwargs):
# super().__init__(**kwargs) |