Update README.md
Browse files
README.md
CHANGED
@@ -41,7 +41,7 @@ If you use this model, please cite the following paper:
|
|
41 |
|
42 |
## Usage
|
43 |
|
44 |
-
|
45 |
from transformers import GPTNeoXForCausalLM, GPTNeoXTokenizerFast
|
46 |
|
47 |
model = GPTNeoXForCausalLM.from_pretrained("NYTK/PULI-GPT-3SX")
|
@@ -61,4 +61,4 @@ gen_tokens = model.generate(
|
|
61 |
|
62 |
gen_text = tokenizer.batch_decode(gen_tokens)[0]
|
63 |
print(gen_text)
|
64 |
-
|
|
|
41 |
|
42 |
## Usage
|
43 |
|
44 |
+
```python
|
45 |
from transformers import GPTNeoXForCausalLM, GPTNeoXTokenizerFast
|
46 |
|
47 |
model = GPTNeoXForCausalLM.from_pretrained("NYTK/PULI-GPT-3SX")
|
|
|
61 |
|
62 |
gen_text = tokenizer.batch_decode(gen_tokens)[0]
|
63 |
print(gen_text)
|
64 |
+
```
|