Hackoor commited on
Commit
c9c6516
·
1 Parent(s): 2c53f01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def create_conversational_chain(vector_store):
60
  load_dotenv()
61
  # Create llm
62
  llm = CTransformers(streaming=True,
63
- model="https://huggingface.co/conceptofmind/Yarn-Llama-2-13b-128k",
64
  callbacks=[StreamingStdOutCallbackHandler()],
65
  model_type="llama", config={'max_new_tokens': 500, 'temperature': 0.01})
66
  # llm = Replicate(
 
60
  load_dotenv()
61
  # Create llm
62
  llm = CTransformers(streaming=True,
63
+ model = AutoModel.from_pretrained("NousResearch/Nous-Hermes-llama-2-7b"),
64
  callbacks=[StreamingStdOutCallbackHandler()],
65
  model_type="llama", config={'max_new_tokens': 500, 'temperature': 0.01})
66
  # llm = Replicate(