moriire commited on
Commit
3ace823
1 Parent(s): 71778ca

Update app/llm.py

Browse files
Files changed (1) hide show
  1. app/llm.py +3 -3
app/llm.py CHANGED
@@ -38,9 +38,9 @@ llm_chat = llama_cpp.Llama.from_pretrained(
38
  #chat_format="llama-2"
39
  )
40
  llm_generate = llama_cpp.Llama.from_pretrained(
41
- repo_id="moriire/healthcare-ai-q2_k",
42
- filename="*.gguf",
43
- tokenizer=llama_cpp.llama_tokenizer.LlamaHFTokenizer.from_pretrained("moriire/healthcare-ai-adapter-merged_16bit"),
44
  verbose=False,
45
  n_ctx=4096,
46
  n_gpu_layers=0,
 
38
  #chat_format="llama-2"
39
  )
40
  llm_generate = llama_cpp.Llama.from_pretrained(
41
+ repo_id="Qwen/Qwen1.5-0.5B-Chat-GGUF",
42
+ filename="*q4_0.gguf",
43
+ #tokenizer=llama_cpp.llama_tokenizer.LlamaHFTokenizer.from_pretrained("moriire/healthcare-ai-adapter-merged_16bit"),
44
  verbose=False,
45
  n_ctx=4096,
46
  n_gpu_layers=0,