Elbachaik commited on
Commit
7f79530
·
verified ·
1 Parent(s): 636a003

Update rag_retriever.py

Browse files
Files changed (1) hide show
  1. rag_retriever.py +1 -1
rag_retriever.py CHANGED
@@ -17,7 +17,7 @@ def initialize_llm():
17
  load_dotenv()
18
  HF_TOKEN = os.getenv("HUGGINGFACE_API_KEY")
19
 
20
- Settings.llm = HuggingFaceInferenceAPI(model_name="mistralai/Mixtral-8x7B-Instruct-v0.1", model_kwargs={"temperature": 0.1}, token=HF_TOKEN)
21
  Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5")
22
 
23
 
 
17
  load_dotenv()
18
  HF_TOKEN = os.getenv("HUGGINGFACE_API_KEY")
19
 
20
+ Settings.llm = HuggingFaceInferenceAPI(model_name="HuggingFaceH4/zephyr-7b-alpha", token=HUGGINGFACE_API_KEY)
21
  Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5")
22
 
23