Add huggingface model support

2024-10-15 19:40:08 +08:00
parent 2d0d25005b
commit 1e78068268
11 changed files with 100 additions and 56 deletions
--- a/reproduce/Step_3.py
+++ b/reproduce/Step_3.py
@@ -52,7 +52,7 @@ def run_queries_and_save_to_json(queries, rag_instance, query_param, output_file

 if __name__ == "__main__":
    cls = "agriculture"
-    mode = "hybird"
+    mode = "hybrid"
    WORKING_DIR = "../{cls}"

    rag = LightRAG(working_dir=WORKING_DIR)