baidu
/

ERNIE-4.5-300B-A47B-PT

Text Generation

Model card Files Files and versions

AntonV HF Staff commited on Jul 21

Commit

0e63afd

·

1 Parent(s): 0d456cf

rm remote

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -76,8 +76,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_name = "baidu/ERNIE-4.5-300B-A47B-PT"
 # load the tokenizer and the model
-tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
 # prepare the model input
 prompt = "Give me a short introduction to large language model."

 model_name = "baidu/ERNIE-4.5-300B-A47B-PT"
 # load the tokenizer and the model
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 # prepare the model input
 prompt = "Give me a short introduction to large language model."