fragata commited on
Commit
e9e8268
1 Parent(s): d520d02

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -1
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
+ '''