Update README.md
Browse files
README.md
CHANGED
@@ -37,4 +37,28 @@ If you use this model, please cite the following paper:
|
|
37 |
pages = {0}
|
38 |
}
|
39 |
|
40 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
pages = {0}
|
38 |
}
|
39 |
|
40 |
+
```
|
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(
|
56 |
+
input_ids,
|
57 |
+
do_sample=True,
|
58 |
+
temperature=0.9,
|
59 |
+
max_length=100,
|
60 |
+
)
|
61 |
+
|
62 |
+
gen_text = tokenizer.batch_decode(gen_tokens)[0]
|
63 |
+
print(gen_text)
|
64 |
+
'''
|