jbilcke-hf HF staff commited on
Commit
0409ff4
1 Parent(s): 8df21b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -153,7 +153,7 @@ LLM_STOP_WORDS= ["</s>","<|user|>","/s>"]
153
  LLAMA_VERBOSE=False
154
 
155
  print("Running LLM Zephyr")
156
- llm_zephyr = Llama(model_path=zephyr_model_path,n_gpu_layers=GPU_LAYERS-10,max_new_tokens=256, context_window=4096, n_ctx=4096,n_batch=128,verbose=LLAMA_VERBOSE)
157
 
158
  def split_sentences(text, max_len):
159
  # Apply custom rules to enforce sentence breaks with double punctuation
 
153
  LLAMA_VERBOSE=False
154
 
155
  print("Running LLM Zephyr")
156
+ llm_zephyr = Llama(model_path=zephyr_model_path,n_gpu_layers=GPU_LAYERS-10,max_new_tokens=512, context_window=4096, n_ctx=4096,n_batch=128,verbose=LLAMA_VERBOSE)
157
 
158
  def split_sentences(text, max_len):
159
  # Apply custom rules to enforce sentence breaks with double punctuation