Update app.py
Browse files
app.py
CHANGED
|
@@ -23,9 +23,9 @@ def load_model(model_name):
|
|
| 23 |
raise FileNotFoundError(f"Model file not found at {model_path}")
|
| 24 |
llm = Llama(
|
| 25 |
model_path=model_path,
|
| 26 |
-
n_ctx=
|
| 27 |
n_threads=2,
|
| 28 |
-
n_batch=
|
| 29 |
verbose=False
|
| 30 |
)
|
| 31 |
current_model = model_name
|
|
|
|
| 23 |
raise FileNotFoundError(f"Model file not found at {model_path}")
|
| 24 |
llm = Llama(
|
| 25 |
model_path=model_path,
|
| 26 |
+
n_ctx=1024,
|
| 27 |
n_threads=2,
|
| 28 |
+
n_batch=256,
|
| 29 |
verbose=False
|
| 30 |
)
|
| 31 |
current_model = model_name
|