Update README.md

README.md

@@ -61,8 +61,8 @@ We would like to take this opportunity to thank
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModelForCausalLM.from_pretrained("Local-Novel-LLM-project/Ninja-v1-128k", trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained("Local-Novel-LLM-project/Ninja-v1-128k")
+model = AutoModelForCausalLM.from_pretrained("Local-Novel-LLM-project/Ninja-v1")
+tokenizer = AutoTokenizer.from_pretrained("Local-Novel-LLM-project/Ninja-v1")
 
 prompt = "Once upon a time,"
 input_ids = tokenizer.encode(prompt, return_tensors="pt")