refine readme (#170)

2024-03-29 14:38:15 +08:00
parent 05298d5626
commit e876f58b4c
3 changed files with 15 additions and 15 deletions
--- a/rag/llm/embedding_model.py
+++ b/rag/llm/embedding_model.py
@@ -51,7 +51,7 @@ class Base(ABC):


 class HuEmbedding(Base):
-    def __init__(self, **kwargs):
+    def __init__(self, *args, **kwargs):
        """
        If you have trouble downloading HuggingFace models, -_^ this might help!!