crumb
/

gzip-openhermes

Model card Files Files and versions Community

crumb commited on Nov 20, 2023

Commit

e46a2d1

·

1 Parent(s): 961d21c

Upload model

Files changed (1) hide show

modeling_gzipembed.py +1 -1

modeling_gzipembed.py CHANGED Viewed

@@ -34,7 +34,7 @@ class GZIPEmbeddingModel(PreTrainedModel):
             return self.reduction_head(x)
         return x if not return_tensor else torch.tensor(x)
-    def ncd_r(self,rp):
         i=r[0]
         return i,self.ncd(r[1],r[2])

             return self.reduction_head(x)
         return x if not return_tensor else torch.tensor(x)
+    def ncd_r(self,r):
         i=r[0]
         return i,self.ncd(r[1],r[2])