Update README.md
Browse files
README.md
CHANGED
@@ -41,15 +41,12 @@ 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")
|
48 |
-
|
49 |
tokenizer = GPTNeoXTokenizerFast.from_pretrained("NYTK/PULI-GPT-3SX")
|
50 |
-
|
51 |
prompt = "Elmesélek egy történetet a nyelvtechnológiáról."
|
52 |
-
|
53 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
|
54 |
|
55 |
gen_tokens = model.generate(
|
|
|
41 |
|
42 |
## Usage
|
43 |
|
44 |
+
```
|
45 |
from transformers import GPTNeoXForCausalLM, GPTNeoXTokenizerFast
|
46 |
|
47 |
model = GPTNeoXForCausalLM.from_pretrained("NYTK/PULI-GPT-3SX")
|
|
|
48 |
tokenizer = GPTNeoXTokenizerFast.from_pretrained("NYTK/PULI-GPT-3SX")
|
|
|
49 |
prompt = "Elmesélek egy történetet a nyelvtechnológiáról."
|
|
|
50 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
|
51 |
|
52 |
gen_tokens = model.generate(
|