fixed model loading issue (openai#66)

wangjunxiao · Mar 23, 2021 · 8a665a6 · 8a665a6
1 parent 290ac5c
commit 8a665a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clip/model.py b/clip/model.py
@@ -288,7 +288,7 @@ def __init__(self,
         self.ln_final = LayerNorm(transformer_width)
 
         self.text_projection = nn.Parameter(torch.empty(transformer_width, embed_dim))
-        self.logit_scale = nn.Parameter(torch.FloatTensor([np.log(1/0.07)]))
+        self.logit_scale = nn.Parameter(torch.ones([]) * np.log(1 / 0.07))
 
         self.initialize_parameters()