alirezamsh
commited on
Commit
•
20fd94f
1
Parent(s):
1bceb72
Update README.md
Browse files
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-
|
62 |
-
tokenizer = AutoTokenizer.from_pretrained("leeroo/LeerooDedicated-Math-
|
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"
|