Update README.md

README.md

@@ -67,7 +67,7 @@ We would like to take this opportunity to thank
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-model_id = "Local-Novel-LLM-project/Ninja-v1-NSFW-128k"
+model_id = "Local-Novel-LLM-project/Ninja-v1-NSFW"
 new_tokens = 1024
 
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, attn_implementation="flash_attention_2", device_map="auto")