From 312ae33f824b5305689eb6135b161f3c73d46010 Mon Sep 17 00:00:00 2001 From: "Aivin V. Solatorio" Date: Sun, 25 Feb 2024 23:52:49 -0500 Subject: [PATCH] Fix collection name (#29) Signed-off-by: Aivin V. Solatorio --- llm4data/embeddings/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llm4data/embeddings/base.py b/llm4data/embeddings/base.py index d319608..f05e8cc 100644 --- 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)