TaruniSwathi commited on
Commit
2f514a4
·
verified ·
1 Parent(s): 8da6b91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -28,11 +28,12 @@ print("Loading GGUF model...")
28
 
29
  llm = Llama(
30
  model_path=model_path,
31
- n_ctx=4096,
32
  n_threads=max(1, os.cpu_count() or 2),
33
  n_threads_batch=max(1, os.cpu_count() or 2),
34
- n_batch=128,
35
- n_gpu_layers=1,
 
36
  verbose=False,
37
  )
38
 
@@ -257,4 +258,8 @@ if __name__ == "__main__":
257
  demo.queue(
258
  default_concurrency_limit=1,
259
  max_size=10,
260
- ).launch()
 
 
 
 
 
28
 
29
  llm = Llama(
30
  model_path=model_path,
31
+ n_ctx=2048,
32
  n_threads=max(1, os.cpu_count() or 2),
33
  n_threads_batch=max(1, os.cpu_count() or 2),
34
+ n_batch=64,
35
+ n_gpu_layers=0,
36
+ use_mmap=True,
37
  verbose=False,
38
  )
39
 
 
258
  demo.queue(
259
  default_concurrency_limit=1,
260
  max_size=10,
261
+ ).launch(
262
+ server_name="0.0.0.0",
263
+ server_port=7860,
264
+ show_error=True,
265
+ )