crumb
/

gzip-openhermes

Model card Files Files and versions

crumb commited on Nov 20, 2023

Commit

83654d1

·

1 Parent(s): f6a8bfb

Upload model

Files changed (1) hide show

modeling_gzipembed.py +1 -1

modeling_gzipembed.py CHANGED Viewed

@@ -16,7 +16,7 @@ class GZIPEmbeddingModel(PreTrainedModel):
             self.reduction_head = None
             self.dummy_parameter = torch.nn.Parameter(torch.ones(1))
-    def forward(self, prompt, num_procs=16):
         global calculate_ncd_row
         global p
         def calculate_ncd_row(data_row):

             self.reduction_head = None
             self.dummy_parameter = torch.nn.Parameter(torch.ones(1))
+    def forward(self, prompt, num_procs=16, return_tensor=True):
         global calculate_ncd_row
         global p
         def calculate_ncd_row(data_row):