SwatGarg commited on
Commit
06ee85e
·
verified ·
1 Parent(s): a17d271

Update generator.py

Browse files
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
3
 
4
- def load_llm(repo_id="HuggingFaceH4/zephyr-7b-beta"):
5
  token = os.getenv("HUGGINGFACE_API_TOKEN")
6
 
7
  # 1. Initialize the endpoint
 
1
  import os
2
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
3
 
4
+ def load_llm(repo_id="microsoft/Phi-3-mini-4k-instruct"):
5
  token = os.getenv("HUGGINGFACE_API_TOKEN")
6
 
7
  # 1. Initialize the endpoint