Commit
•
c644647
1
Parent(s):
7d8f63e
Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,6 @@ tokenizer.add_special_tokens({'pad_token': '[PAD]'})
|
|
| 31 |
tokenizer.max_len = 512
|
| 32 |
|
| 33 |
# load the fine-tuned model
|
| 34 |
-
import torch
|
| 35 |
model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion')
|
| 36 |
|
| 37 |
# generate text using fine-tuned model
|
|
|
|
| 31 |
tokenizer.max_len = 512
|
| 32 |
|
| 33 |
# load the fine-tuned model
|
|
|
|
| 34 |
model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion')
|
| 35 |
|
| 36 |
# generate text using fine-tuned model
|