crumb commited on
Commit
533b56a
·
1 Parent(s): e46a2d1

Upload model

Browse files
Files changed (1) hide show
  1. modeling_gzipembed.py +2 -2
modeling_gzipembed.py CHANGED
@@ -35,8 +35,8 @@ class GZIPEmbeddingModel(PreTrainedModel):
35
  return x if not return_tensor else torch.tensor(x)
36
 
37
  def ncd_r(self,r):
38
- i=r[0]
39
- return i,self.ncd(r[1],r[2])
40
 
41
  def normalize(self, x):
42
  x = ''.join([char for char in x.lower() if char in "abcdefghijklmnopqrstuvwxyz "])
 
35
  return x if not return_tensor else torch.tensor(x)
36
 
37
  def ncd_r(self,r):
38
+ i=r[0][0]
39
+ return i,self.ncd(r[0][1],r[1])
40
 
41
  def normalize(self, x):
42
  x = ''.join([char for char in x.lower() if char in "abcdefghijklmnopqrstuvwxyz "])