Update README.md

README.md

@@ -41,14 +41,12 @@ We will continue to release improved versions of Aquila model as open source. Fo
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-from cyg_conversation import covert_prompt_to_input_ids_with_history
 
-tokenizer = AutoTokenizer.from_pretrained("BAAI/Aquila-7B")
-model = AutoModelForCausalLM.from_pretrained("BAAI/Aquila-7B")
+model_info = "BAAI/Aquila-7B"
+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:0")
-vocab = tokenizer.vocab
-print(len(vocab))
 
 text = "汽车EDR是什么"