openai-community/gpt2 · Model does not work with device set to `mps`

Model does not work with device set to `mps`

by akbir - opened Jun 8, 2022

akbir

Jun 8, 2022

•

Inference on M1 GPU (device=mps) does not work.

Shapes end up being wrong here, not sure if this is a pytorch code or bad model implementation.

OpenAI community org Jun 8, 2022

•

Can you share a snippet of code to reproduce?

Aug 26, 2022

This comment has been hidden

Aug 26, 2022

This comment has been hidden

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment