Update README.md
Browse files
README.md
CHANGED
@@ -47,7 +47,7 @@ Then, you can use the model as follows:
|
|
47 |
```python
|
48 |
# pip install transformers
|
49 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
50 |
-
checkpoint = "jinaai/
|
51 |
|
52 |
device = "cuda" # for GPU usage or "cpu" for CPU usage
|
53 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
|
|
47 |
```python
|
48 |
# pip install transformers
|
49 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
50 |
+
checkpoint = "jinaai/reader-lm-1.5b"
|
51 |
|
52 |
device = "cuda" # for GPU usage or "cpu" for CPU usage
|
53 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|