ps1811 commited on
Commit
771aff5
·
verified ·
1 Parent(s): 3cf50bf

Update app/models/llm.py

Browse files
Files changed (1) hide show
  1. 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", "-1"))
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=False,
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)