Merge pull request #1587 from ribeiromiranda/patch-1

Fixed use_cuda issue in compute_embeddings.py
This commit is contained in:
Eren Gölge
2022-05-29 14:51:08 +02:00
committed by GitHub
+1 -1
View File
@@ -110,7 +110,7 @@ class EmbeddingManager(BaseIDManager):
self.load_embeddings_from_file(embedding_file_path)
if encoder_model_path and encoder_config_path:
self.init_encoder(encoder_model_path, encoder_config_path)
self.init_encoder(encoder_model_path, encoder_config_path, use_cuda)
@property
def embedding_dim(self):