FredZhang7
/

distilgpt2-stable-diffusion

Text Generation

stable-diffusion

prompt-generator

text-generation-inference

Model card Files Files and versions

FredZhang7 commited on Dec 6, 2022

Commit

c4330c7

·

1 Parent(s): 17b2af5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -37,7 +37,7 @@ tokenizer.max_len = 512
 # load the fine-tuned model
 import torch
 model = GPT2LMHeadModel.from_pretrained('distilgpt2')
-model.load_state_dict(torch.load('model.pt'))
 # generate text using fine-tuned model
 from transformers import pipeline

 # load the fine-tuned model
 import torch
 model = GPT2LMHeadModel.from_pretrained('distilgpt2')
+model.load_state_dict(torch.load('distil-sd-gpt2.pt.pt'))
 # generate text using fine-tuned model
 from transformers import pipeline