Spaces:

3laa2
/

Text2img

Runtime error

3laa2 commited on Jun 8, 2023

Commit

fa3d600

1 Parent(s): 3df1457

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,6 +16,7 @@ def tok_mod():
   tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
   tokenizer.add_special_tokens({'pad_token': '[PAD]'})
   model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion-v2')
   return model,tokenizer

   tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
   tokenizer.add_special_tokens({'pad_token': '[PAD]'})
   model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion-v2')
+  model.to('cpu')
   return model,tokenizer