sugiv commited on
Commit
564c266
·
1 Parent(s): d24e63f

Change of settings again

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ REPO_ID = "sugiv/leetmonkey-peft-gguf"
46
  model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_NAME, cache_dir="./models")
47
  #llm = Llama(model_path=model_path, n_ctx=1024, n_threads=8, n_gpu_layers=-1, verbose=False, mlock=True) ## TPU
48
  #llm = Llama(model_path=model_path, n_ctx=1024, n_threads=16, n_gpu_layers=0, verbose=False, mlock=False) ## CPU only
49
- llm = Llama(model_path=model_path, n_ctx=1024, n_threads=16, n_gpu_layers=10, verbose=False, mlock=False) ## Nvidia
50
  logger.info("8-bit model loaded successfully")
51
 
52
  # User data storage
 
46
  model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_NAME, cache_dir="./models")
47
  #llm = Llama(model_path=model_path, n_ctx=1024, n_threads=8, n_gpu_layers=-1, verbose=False, mlock=True) ## TPU
48
  #llm = Llama(model_path=model_path, n_ctx=1024, n_threads=16, n_gpu_layers=0, verbose=False, mlock=False) ## CPU only
49
+ llm = Llama(model_path=model_path, n_ctx=1024, n_threads=16, n_gpu_layers=-1, verbose=False, mlock=False) ## Nvidia
50
  logger.info("8-bit model loaded successfully")
51
 
52
  # User data storage