File size: 555 Bytes
0102396 b53ebf8 0102396 dd2b0df b53ebf8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
---
license: mit
language:
- ku
base_model: openai-community/gpt2
pipeline_tag: text-generation
library_name: transformers
---
use this :<br/>
<br/>
from transformers import pipeline,GPT2Tokenizer<br/>
test_input = "حەزم"<br/>
tokenizer = GPT2Tokenizer.from_pretrained("bayandorian/gpt2-kurdish")<br/>
generator = pipeline('text-generation', model='bayandorian/gpt2-kurdish',device="cuda")<br/>
sample = generator(test_input, temperature=1.2, top_k=50, top_p=0.9, repetition_penalty=1.2, pad_token_id = tokenizer.eos_token_id)<br/>
print(sample)<br/> |