alirezamsh commited on
Commit
20fd94f
1 Parent(s): 1bceb72

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -58,8 +58,8 @@ You can also add your OpenAI API to get the full answer when `<GPT4>` token is g
58
  from openai import OpenAI
59
  from transformers import AutoModelForCausalLM, AutoTokenizer
60
 
61
- model = AutoModelForCausalLM.from_pretrained("leeroo/LeerooDedicated-Math-7B-OOE", trust_remote_code=True)
62
- tokenizer = AutoTokenizer.from_pretrained("leeroo/LeerooDedicated-Math-7B-OOE")
63
  openai_client = OpenAI(
64
  api_key= "OPENAI_API_KEY",
65
  base_url= "https://api.openai.com/v1"
 
58
  from openai import OpenAI
59
  from transformers import AutoModelForCausalLM, AutoTokenizer
60
 
61
+ model = AutoModelForCausalLM.from_pretrained("leeroo/LeerooDedicated-Math-7b", trust_remote_code=True)
62
+ tokenizer = AutoTokenizer.from_pretrained("leeroo/LeerooDedicated-Math-7b")
63
  openai_client = OpenAI(
64
  api_key= "OPENAI_API_KEY",
65
  base_url= "https://api.openai.com/v1"