towhee.models.clip.clip.create_model

towhee.models.clip.clip.create_model(model_name: Optional[str] = None, pretrained: bool = False, weights_path: Optional[str] = None, jit: bool = False, device: Optional[str] = None, **kwargs) → CLIP

Create a CLIP model. :param model_name: CLIP model name, can be one of ‘clip_resnet_r50’, ‘clip_resnet_r101’, ‘clip_vit_b16’, ‘clip_vit_b32’ :type model_name: str :param pretrained: Whether to load pretrained weights. :type pretrained: bool :param weights_path: Path to the weights file. :type weights_path: str :param jit: Whether returned one is a jit model, only useful when pretrained is True. :type jit: bool :param device: Model device to use. :type device: str :param **kwargs: Extra arguments to pass to the model. :type **kwargs: dict

Returns:

The CLIP model.

Return type:

model (CLIP)

>>> from towhee.models import clip
>>> model = clip.create_model("clip_resnet_r50")
>>> model.__class__.__name__
'CLIP'