Update app.py
Browse files
app.py
CHANGED
@@ -16,6 +16,7 @@ def tok_mod():
|
|
16 |
tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
|
17 |
tokenizer.add_special_tokens({'pad_token': '[PAD]'})
|
18 |
model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion-v2')
|
|
|
19 |
return model,tokenizer
|
20 |
|
21 |
|
|
|
16 |
tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
|
17 |
tokenizer.add_special_tokens({'pad_token': '[PAD]'})
|
18 |
model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion-v2')
|
19 |
+
model.to('cpu')
|
20 |
return model,tokenizer
|
21 |
|
22 |
|