Commit
·
b7a7f6d
1
Parent(s):
d4945a6
Update README.md
Browse files
README.md
CHANGED
|
@@ -74,7 +74,6 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
|
| 74 |
tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
|
| 75 |
tokenizer.add_special_tokens({'pad_token': '[PAD]'})
|
| 76 |
model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion-v2')
|
| 77 |
-
model.eval()
|
| 78 |
|
| 79 |
prompt = r'a cat sitting' # the beginning of the prompt
|
| 80 |
temperature = 0.9 # a higher temperature will produce more diverse results, but with a higher risk of less coherent text.
|
|
|
|
| 74 |
tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
|
| 75 |
tokenizer.add_special_tokens({'pad_token': '[PAD]'})
|
| 76 |
model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion-v2')
|
|
|
|
| 77 |
|
| 78 |
prompt = r'a cat sitting' # the beginning of the prompt
|
| 79 |
temperature = 0.9 # a higher temperature will produce more diverse results, but with a higher risk of less coherent text.
|