aadya1762 commited on
Commit
e96d38d
·
1 Parent(s): 603cefc

bug fixes

Browse files
Files changed (1) hide show
  1. gemmademo/_model.py +1 -1
gemmademo/_model.py CHANGED
@@ -111,7 +111,7 @@ class LlamaCppGemmaModel:
111
  n_gpu_layers=n_gpu_layers,
112
  n_batch=8,
113
  verbose=False,
114
- chat_format="chatml",
115
  )
116
 
117
  self.messages.append(_sys_prompt)
 
111
  n_gpu_layers=n_gpu_layers,
112
  n_batch=8,
113
  verbose=False,
114
+ chat_format="gemma",
115
  )
116
 
117
  self.messages.append(_sys_prompt)