Update README.md
Browse files
README.md
CHANGED
|
@@ -57,8 +57,8 @@ We would like to take this opportunity to thank
|
|
| 57 |
```python
|
| 58 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 59 |
|
| 60 |
-
model = AutoModelForCausalLM.from_pretrained("Local-Novel-LLM-project/
|
| 61 |
-
tokenizer = AutoTokenizer.from_pretrained("Local-Novel-LLM-project/
|
| 62 |
|
| 63 |
prompt = "Once upon a time,"
|
| 64 |
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|
|
|
|
| 57 |
```python
|
| 58 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 59 |
|
| 60 |
+
model = AutoModelForCausalLM.from_pretrained("Local-Novel-LLM-project/Vecteus-v1", trust_remote_code=True)
|
| 61 |
+
tokenizer = AutoTokenizer.from_pretrained("Local-Novel-LLM-project/Vecteus-v1")
|
| 62 |
|
| 63 |
prompt = "Once upon a time,"
|
| 64 |
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|