File size: 738 Bytes
753a872 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import copy
import transformers
from transformers import PretrainedConfig, Qwen2Config
from .configuration_aimv2 import AIMv2Config
class FlashVLDynamicISSConfig(PretrainedConfig):
model_type = 'FlashVLDynamicISSConfig'
is_composition = True
def __init__(
self,
vision_config,
llm_config,
**kwargs
):
super().__init__(**kwargs)
self.vision_config = AIMv2Config(**vision_config)
self.llm_config = Qwen2Config(**llm_config)
def to_dict(self):
output = copy.deepcopy(self.__dict__)
output['vision_config'] = self.vision_config.to_dict()
output['llm_config'] = self.llm_config.to_dict()
return output
|