sugiv commited on
Commit
ca0a740
·
1 Parent(s): cc60d5c

Change of settings again for CPU, trying 16 CPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ REPO_ID = "sugiv/leetmonkey-peft-gguf"
45
  # Load the model
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=2048, n_threads=16, n_gpu_layers=-1, verbose=False, mlock=True) ## TPU
48
- llm = Llama(model_path=model_path, n_ctx=1024, n_threads=8, n_gpu_layers=0, verbose=False, mlock=False) ## CPU only
49
  #llm = Llama(model_path=model_path, n_ctx=1024, n_threads=8, n_gpu_layers=-1, verbose=False, mlock=False) ## Nvidia
50
  logger.info("8-bit model loaded successfully")
51
 
 
45
  # Load the model
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=2048, n_threads=16, n_gpu_layers=-1, verbose=False, mlock=True) ## TPU
48
+ llm = Llama(model_path=model_path, n_ctx=1024, n_threads=2, n_gpu_layers=0, verbose=False, mlock=False) ## CPU only
49
  #llm = Llama(model_path=model_path, n_ctx=1024, n_threads=8, n_gpu_layers=-1, verbose=False, mlock=False) ## Nvidia
50
  logger.info("8-bit model loaded successfully")
51