model_config.py

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)