Rename VisualTransformer -> VisionTransformer (openai#97)

Fixes openai#94
KoolenDasheppi · Jul 19, 2021 · ea41722 · ea41722
1 parent 504cf9e
commit ea41722
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/clip/model.py b/clip/model.py
@@ -199,7 +199,7 @@ def forward(self, x: torch.Tensor):
         return self.resblocks(x)
 
 
-class VisualTransformer(nn.Module):
+class VisionTransformer(nn.Module):
     def __init__(self, input_resolution: int, patch_size: int, width: int, layers: int, heads: int, output_dim: int):
         super().__init__()
         self.input_resolution = input_resolution
@@ -266,7 +266,7 @@ def __init__(self,
             )
         else:
             vision_heads = vision_width // 64
-            self.visual = VisualTransformer(
+            self.visual = VisionTransformer(
                 input_resolution=image_resolution,
                 patch_size=vision_patch_size,
                 width=vision_width,