Update app/models/llm.py
Browse files- app/models/llm.py +2 -2
app/models/llm.py
CHANGED
|
@@ -20,14 +20,14 @@ def load_model():
|
|
| 20 |
|
| 21 |
print(f"✅ [load_model] model downloaded at {model_path}", flush=True)
|
| 22 |
|
| 23 |
-
gpu_layers = int(os.getenv("LLAMA_GPU_LAYERS", "
|
| 24 |
print(f"🚀 [load_model] initializing Llama with n_gpu_layers={gpu_layers}", flush=True)
|
| 25 |
|
| 26 |
_model = Llama(
|
| 27 |
model_path=model_path,
|
| 28 |
n_ctx=4096,
|
| 29 |
n_gpu_layers=gpu_layers,
|
| 30 |
-
verbose=
|
| 31 |
)
|
| 32 |
|
| 33 |
print("✅ [load_model] model initialized", flush=True)
|
|
|
|
| 20 |
|
| 21 |
print(f"✅ [load_model] model downloaded at {model_path}", flush=True)
|
| 22 |
|
| 23 |
+
gpu_layers = int(os.getenv("LLAMA_GPU_LAYERS", "0"))
|
| 24 |
print(f"🚀 [load_model] initializing Llama with n_gpu_layers={gpu_layers}", flush=True)
|
| 25 |
|
| 26 |
_model = Llama(
|
| 27 |
model_path=model_path,
|
| 28 |
n_ctx=4096,
|
| 29 |
n_gpu_layers=gpu_layers,
|
| 30 |
+
verbose=True,
|
| 31 |
)
|
| 32 |
|
| 33 |
print("✅ [load_model] model initialized", flush=True)
|