| import transformers | |
| class MCLIPConfig(transformers.PretrainedConfig): | |
| model_type = "M-CLIP" | |
| def __init__(self, modelBase='sentence-transformers/LaBSE', transformerDimSize=768, imageDimSize=768, **kwargs): | |
| self.transformerDimensions = transformerDimSize | |
| self.numDims = imageDimSize | |
| self.modelBase = modelBase | |
| super().__init__(**kwargs) | |