amancod commited on
Commit
cde03ab
·
verified ·
1 Parent(s): e986e56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ Follow us [Gathnex](https://medium.com/@gathnex), [linkedin](https://www.linkedi
6
  """
7
 
8
  llm = AutoModelForCausalLM.from_pretrained("llama-2-7b-chat.Q4_K_S.gguf",
9
- model_type='mistral',
10
  max_new_tokens = 1096,
11
  threads = 3,
12
  )
 
6
  """
7
 
8
  llm = AutoModelForCausalLM.from_pretrained("llama-2-7b-chat.Q4_K_S.gguf",
9
+ model_type='llama',
10
  max_new_tokens = 1096,
11
  threads = 3,
12
  )