Update README.md
Browse files
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.
|
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")
|