Fixing a syntax problem in SimilarityHead.

Author: rodgzilla

model_pytorch.py

@@ -250,7 +250,7 @@ def __init__(self, clf_token, cfg):
         self.n_embd = cfg.n_embd
         self.clf_token = clf_token
         self.dropout = nn.Dropout(cfg.clf_pdrop)
-        self.linear = nn.Linear(cfg_n_embd, 1)
+        self.linear = nn.Linear(cfg.n_embd, 1)
 
         nn.init.normal_(self.linear.weight, std = 0.02)
         nn.init.normal_(self.linear.bias, 0)