hsuwill000 commited on
Commit
0d3a8e6
Β·
verified Β·
1 Parent(s): a9e6e4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,9 +39,9 @@ def setup_model(model_id):
39
  gguf_path = f"{local_dir}.gguf"
40
  if not os.path.exists(gguf_path):
41
  run_command(f'python llama.cpp/convert_hf_to_gguf.py ./{local_dir} --outfile {gguf_path}')
42
- quantized_path = f"{local_dir}-Q2_K.gguf"
43
  if not os.path.exists(quantized_path):
44
- run_command(f'./llama.cpp/build/bin/llama-quantize ./{gguf_path} {quantized_path} Q2_K')
45
  return quantized_path
46
 
47
  def chat_with_model(message, history, system_prompt, temperature, max_tokens, top_k, top_p):
 
39
  gguf_path = f"{local_dir}.gguf"
40
  if not os.path.exists(gguf_path):
41
  run_command(f'python llama.cpp/convert_hf_to_gguf.py ./{local_dir} --outfile {gguf_path}')
42
+ quantized_path = f"{local_dir}-IQ2_XXS.gguf"
43
  if not os.path.exists(quantized_path):
44
+ run_command(f'./llama.cpp/build/bin/llama-quantize ./{gguf_path} {quantized_path} IQ2_XXS')
45
  return quantized_path
46
 
47
  def chat_with_model(message, history, system_prompt, temperature, max_tokens, top_k, top_p):