alirezamsh commited on
Commit
1cc2a7f
·
verified ·
1 Parent(s): 948ab7d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -18,8 +18,8 @@ In evaluations using the GSM8k dataset of [OpenLLM Leaderboard](https://huggingf
18
  ```python
19
  from transformers import AutoModelForCausalLM, AutoTokenizer
20
 
21
- model = AutoModelForCausalLM.from_pretrained("leeroo/LeerooDedicated-Math", trust_remote_code=True)
22
- tokenizer = AutoTokenizer.from_pretrained("leeroo/LeerooDedicated-Math")
23
  device = model.device
24
 
25
  # the following question is answered by the leeroo expert
 
18
  ```python
19
  from transformers import AutoModelForCausalLM, AutoTokenizer
20
 
21
+ model = AutoModelForCausalLM.from_pretrained("leeroo/LeerooDedicated-Math-7b", trust_remote_code=True)
22
+ tokenizer = AutoTokenizer.from_pretrained("leeroo/LeerooDedicated-Math-7b")
23
  device = model.device
24
 
25
  # the following question is answered by the leeroo expert