Fawaz0ibra commited on
Commit
933c4ee
·
verified ·
1 Parent(s): b799330

Update chain_setup.py

Browse files
Files changed (1) hide show
  1. chain_setup.py +5 -4
chain_setup.py CHANGED
@@ -17,11 +17,12 @@ def load_llm():
17
  llm = LlamaCpp(
18
  model_path=model_file,
19
  flash_attn=False,
20
- n_ctx=2048, # or 4096
21
- n_batch=512, # or even 256
22
- chat_format='chatml'
 
23
  )
24
-
25
  return llm
26
 
27
  def build_conversational_chain(vectorstore):
 
17
  llm = LlamaCpp(
18
  model_path=model_file,
19
  flash_attn=False,
20
+ n_ctx=2048,
21
+ n_batch=512,
22
+ chat_format="chatml",
23
+ grammar="" # set grammar to an empty string to bypass template parsing issues
24
  )
25
+
26
  return llm
27
 
28
  def build_conversational_chain(vectorstore):