Commit
•
0409ff4
1
Parent(s):
8df21b1
Update app.py
Browse files
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=
|
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
|