acbdkk commited on
Commit
88c6be7
·
verified ·
1 Parent(s): b332c87

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -15,8 +15,8 @@ model_id = "acbdkk/SupaMATH3"
15
  device = "cpu"
16
 
17
 
18
- model = AutoModelForCausalLM.from_pretrained(model_id, model_type="llama", lib="avx2", hf=True)
19
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
20
 
21
  def get_prompt(message: str, chat_history: list[tuple[str, str]],
22
  system_prompt: str) -> str:
 
15
  device = "cpu"
16
 
17
 
18
+ model = AutoModelForCausalLM.from_pretrained(model_id,hf=True)
19
+ tokenizer = AutoTokenizer.from_pretrained("acbdkk/SupaMATH3")
20
 
21
  def get_prompt(message: str, chat_history: list[tuple[str, str]],
22
  system_prompt: str) -> str: