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

Change of settings again for CPU, trying 16 CPU

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,8 +45,8 @@ 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
 
52
  # User data storage
 
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
 
52
  # User data storage