towhee.models.vit.vit.create_model

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

Create ViT model. :param model_name: ViT model name. :type model_name: str :param pretrained: Whether it is a pretrained model. :type pretrained: bool :param weights_path: If pretrained, you can use your local weights. :type weights_path: str :param device: Cuda or cpu. :type device: str

Returns:

VitModel