Views
No views yet
GPT2LMHeadModel for Language Modeling.gpt2 (124M parameters)GPT2LMHeadModel)do_sample=True and temperature=0.8 are set as default generation parameters to encourage creative and diverse outputs.1from transformers import pipeline, set_seed
2
3generator = pipeline(
4 "text-generation",
5 model="[YOUR_HF_USERNAME]/PoeticTextGenerator_GPT2"
6)
7set_seed(42)
8
9# Conditional Generation (Prompting a theme)
10prompt = "The shadow of the moon fell upon the silent street,"
11output = generator(
12 prompt,
13 max_length=50,
14 num_return_sequences=1,
15 temperature=0.9,
16 top_p=0.95,
17 do_sample=True
18)
19print(output[0]['generated_text'])
20
21# Unconditional Generation (Starting from a single word)
22# output = generator("A", max_length=100, num_return_sequences=1)