fixed model loading issue (#66)

2021-03-23 03:05:17 -04:00 · 2021-03-23 03:05:17 -04:00 · 8a665a683d
parent 290ac5cb15
commit 8a665a683d
1 changed files with 1 additions and 1 deletions
--- a/clip/model.py
+++ b/clip/model.py
@ -288,7 +288,7 @@ class CLIP(nn.Module):
        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()