Open dbl001 opened 1 year ago
Same error with: 2.0.0.dev20230211
% pip show torch
Name: torch
Version: 2.0.0.dev20230211
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3
Location: /Users/davidlaxer/anaconda3/envs/AI-Feynman/lib/python3.10/site-packages
Editable project location: /Users/davidlaxer/anaconda3/envs/AI-Feynman/lib/python3.10/site-packages
Requires: filelock, networkx, sympy, typing-extensions
Required-by: accelerate, aifeynman, audiocraft, audiolm-pytorch, benepar, captum, demucs, diffq, dora-search, flashy, fschat, julius, minGPT, openunmix, parallelformers, pytorch-transformers, sentence-transformers, timm, torch-struct, torch-utils, torchaudio, torchtraining-nightly, torchvision, tuned-lens, vector-quantize-pytorch, whisper, xformers
🐛 Describe the bug
I am running the transformer model Pix2StructForConditionalGeneration, Pix2StructProcessor on MacOS 13.4 on an iMac 27" 2020 with an AMD Radeon Pro 5700XT.
The code to run the Transformer is here:
https://huggingface.co/google/pix2struct-ai2d-base
The code runs with 'CPU' but with 'MPS' gets:
Here's the code:
Versions
cc @kulinseth @albanD @malfet @DenisVieriu97 @razarmehr @abhudev