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