umisetokikaze commited on
Commit
e65bef9
·
verified ·
1 Parent(s): 73889e3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -64,8 +64,8 @@ We would like to take this opportunity to thank
64
  ```python
65
  from transformers import AutoModelForCausalLM, AutoTokenizer
66
 
67
- model = AutoModelForCausalLM.from_pretrained("Local-Novel-LLM-project/Ninja-v1-128k", trust_remote_code=True)
68
- tokenizer = AutoTokenizer.from_pretrained("Local-Novel-LLM-project/Ninja-v1-128k")
69
 
70
  prompt = "Once upon a time,"
71
  input_ids = tokenizer.encode(prompt, return_tensors="pt")
 
64
  ```python
65
  from transformers import AutoModelForCausalLM, AutoTokenizer
66
 
67
+ model = AutoModelForCausalLM.from_pretrainedLocal-Novel-LLM-project/Ninja-v1-NSFW-128k", trust_remote_code=True)
68
+ tokenizer = AutoTokenizer.from_pretrained("Local-Novel-LLM-project/Ninja-v1-NSFW-128k")
69
 
70
  prompt = "Once upon a time,"
71
  input_ids = tokenizer.encode(prompt, return_tensors="pt")