Views
No views yet
pip install --upgrade transformers1from transformers import GPT2Tokenizer, GPT2LMHeadModel
2
3# load the pretrained tokenizer
4tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
5tokenizer.add_special_tokens({'pad_token': '[PAD]'})
6tokenizer.max_len = 512
7
8# load the fine-tuned model
9model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion')
10
11# generate text using fine-tuned model
12from transformers import pipeline
13nlp = pipeline('text-generation', model=model, tokenizer=tokenizer)
14ins = "a beautiful city"
15
16# generate 10 samples
17outs = nlp(ins, max_length=80, num_return_sequences=10)
18
19# print the 10 samples
20for i in range(len(outs)):
21 outs[i] = str(outs[i]['generated_text']).replace(' ', '')
22print('\033[96m' + ins + '\033[0m')
23print('\033[93m' + '\n\n'.join(outs) + '\033[0m')