Update app/llm.py
Browse files- 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="
|
| 42 |
-
filename="
|
| 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,
|