File size: 378 Bytes
280e3ed | 1 2 3 4 5 6 7 8 9 10 11 12 | 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)
|