Fix collection name (#29)

Signed-off-by: Aivin V. Solatorio <[email protected]>
worldbank · Feb 26, 2024 · 312ae33 · 312ae33
1 parent a4ab398
commit 312ae33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm4data/embeddings/base.py b/llm4data/embeddings/base.py
@@ -71,7 +71,7 @@ def _common_init(self):
             self.size = self.model_size[self.model_name]
 
         if self.collection_name is None:
-            self.collection_name = f"{self.data_type}_{self.model_name}"
+            self.collection_name = f"{self.data_type}_{self.model_name.replace('/', '_')}"
 
     def _instruct_init(self):
         assert isinstance(self.kwargs, dict)