Update modeling_gzipembed.py

modeling_gzipembed.py

@@ -50,8 +50,10 @@ class GZIPEmbeddingModel(PreTrainedModel):
         Returns:
             `List[np.ndarray]` or `List[tensor]`: List of embeddings for the given sentences
         """
+        import numpy as np
         x = self.forward(sentences, num_procs=batch_size, return_tensor=False)
-        return [torch.tensor(i) for i in x]
+        # return [torch.tensor(i) for i in x]
+        return [np.array(i) for i in x] # test?
 
     def normalize(self, x):
         x = ''.join([char for char in x.lower() if char in "abcdefghijklmnopqrstuvwxyz "])