Views
No views yet
transformers library. Below is an example of how to load and use the model in Python:1from transformers import pipeline, set_seed
2
3generator = pipeline('text-generation', model='agentlans/pythia-70m-wikipedia-paragraphs', device='cuda')
4
5results = generator("Your text here...", max_length=3000, num_return_sequences=3, do_sample=True)
6
7for x in results:
8 print(x['generated_text'] + '\n')max_length: Maximum length of the generated sequence.num_return_sequences: Number of sequences to return.temperature: Controls randomness; lower values make the output more deterministic.top_k and top_p: Controls sampling strategies to influence diversity.set_seed(1234567) to reproduce the results.