moriire commited on
Commit
436d8e0
1 Parent(s): 9c9836d

Update app/llm.py

Browse files
Files changed (1) hide show
  1. app/llm.py +3 -3
app/llm.py CHANGED
@@ -22,7 +22,7 @@ class GenModel(BaseModel):
22
 
23
  class ChatModel(BaseModel):
24
  question: list
25
- system: str = "You are a helpful AI assistant. You are chatting with a {0}. Help as much as you can."
26
  #Also continuously ask for possible symptoms in order to atat a conclusive ailment or sickness and possible solutions.Remember, response in English."
27
  temperature: float = 0.8
28
  seed: int = 101
@@ -32,7 +32,7 @@ class ChatModel(BaseModel):
32
  llm_chat = llama_cpp.Llama.from_pretrained(
33
  repo_id="moriire/healthcare-GGUF",
34
  filename="healthcare-GGUF-unsloth.Q4_K_M.gguf",
35
- tokenizer=llama_cpp.llama_tokenizer.LlamaHFTokenizer.from_pretrained("Qwen/Qwen1.5-0.5B"),
36
  verbose=False,
37
  n_ctx=1024,
38
  n_gpu_layers=0,
@@ -41,7 +41,7 @@ llm_chat = llama_cpp.Llama.from_pretrained(
41
  llm_generate = llama_cpp.Llama.from_pretrained(
42
  repo_id="moriire/healthcare-GGUF",
43
  filename="healthcare-GGUF-unsloth.Q4_K_M.gguf",
44
- tokenizer=llama_cpp.llama_tokenizer.LlamaHFTokenizer.from_pretrained("Qwen/Qwen1.5-0.5B"),
45
  verbose=False,
46
  n_ctx=4096,
47
  n_gpu_layers=0,
 
22
 
23
  class ChatModel(BaseModel):
24
  question: list
25
+ system: str = "You are a helpful AI assistant. You are chatting with a human. Help as much as you can."
26
  #Also continuously ask for possible symptoms in order to atat a conclusive ailment or sickness and possible solutions.Remember, response in English."
27
  temperature: float = 0.8
28
  seed: int = 101
 
32
  llm_chat = llama_cpp.Llama.from_pretrained(
33
  repo_id="moriire/healthcare-GGUF",
34
  filename="healthcare-GGUF-unsloth.Q4_K_M.gguf",
35
+ tokenizer=llama_cpp.llama_tokenizer.LlamaHFTokenizer.from_pretrained("moriire/healthcare-GGUF")
36
  verbose=False,
37
  n_ctx=1024,
38
  n_gpu_layers=0,
 
41
  llm_generate = llama_cpp.Llama.from_pretrained(
42
  repo_id="moriire/healthcare-GGUF",
43
  filename="healthcare-GGUF-unsloth.Q4_K_M.gguf",
44
+ tokenizer=llama_cpp.llama_tokenizer.LlamaHFTokenizer.from_pretrained("moriire/healthcare-GGUF")
45
  verbose=False,
46
  n_ctx=4096,
47
  n_gpu_layers=0,