towhee.models.clip.clipΒΆ

Functions

create_model

Create a CLIP model.

Classes

AttentionPool2d

Attention

Bottleneck

BottleNeck

CLIP

CLIP model

LayerNorm

Subclass torch's LayerNorm to handle fp16.

ModifiedResNet

A ResNet class that is similar to torchvision's but contains the following changes: - There are now 3 "stem" convolutions as opposed to 1, with an average pool instead of a max pool.

QuickGELU

ResidualAttentionBlock

ResidualAttentuonBlock

Transformer

VisionTransformer

ViT