BAAI
/

AquilaCode-multi

Inference Endpoints

Model card Files Files and versions Community

shunxing1234 commited on Jul 21, 2023

Commit

13bf5db

·

1 Parent(s): ef5f0c0

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -58,8 +58,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 model_info = "BAAI/AquilaCode-multi"
-tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True, use_auth_token='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')
-model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True, use_auth_token='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')
 model.eval()
 model.to("cuda:3")

 import torch
 model_info = "BAAI/AquilaCode-multi"
+tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
 model.eval()
 model.to("cuda:3")