Spaces:

ironserengety
/

GraphRAG-Local-to-Global

Sleeping

Saif Rehman Nasir commited on Sep 29, 2024

Commit

0592ad1

1 Parent(s): 80a09fa

Change LLM to Zephyr

Files changed (1) hide show

rag.py CHANGED Viewed

@@ -22,7 +22,7 @@ NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")
 vector_index = os.getenv("VECTOR_INDEX")
 chat_llm = HuggingFaceEndpoint(
-    repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
     task="text-generation",
     max_new_tokens=4096,
     do_sample=False,

 vector_index = os.getenv("VECTOR_INDEX")
 chat_llm = HuggingFaceEndpoint(
+    repo_id="HuggingFaceH4/zephyr-7b-beta",
     task="text-generation",
     max_new_tokens=4096,
     do_sample=False,