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"
|