Saif Rehman Nasir commited on
Commit
c3745cb
·
1 Parent(s): 9618bfc

Revert to llama 3

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -22,7 +22,7 @@ NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")
22
  vector_index = os.getenv("VECTOR_INDEX")
23
 
24
  chat_llm = HuggingFaceEndpoint(
25
- repo_id="HuggingFaceH4/zephyr-7b-beta",
26
  task="text-generation",
27
  max_new_tokens=4096,
28
  do_sample=False,
 
22
  vector_index = os.getenv("VECTOR_INDEX")
23
 
24
  chat_llm = HuggingFaceEndpoint(
25
+ repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
26
  task="text-generation",
27
  max_new_tokens=4096,
28
  do_sample=False,