| from transformers import PretrainedConfig, VisionEncoderDecoderConfig | |
| from typing import List | |
| class Magiv2Config(PretrainedConfig): | |
| model_type = "magiv2" | |
| def __init__( | |
| self, | |
| crop_embedding_model_config: dict = None, | |
| crop_embedding_image_preprocessing_config: dict = None, | |
| **kwargs, | |
| ): | |
| self.kwargs = kwargs | |
| self.crop_embedding_model_config = None | |
| if crop_embedding_model_config is not None: | |
| self.crop_embedding_model_config = PretrainedConfig.from_dict(crop_embedding_model_config) | |
| self.crop_embedding_image_preprocessing_config = crop_embedding_image_preprocessing_config | |
| super().__init__(**kwargs) |