runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/home/user/app/app.py", line 7, in <module> from mario_gpt.lm import MarioLM File "/home/user/app/mario_gpt/lm.py", line 6, in <module> from transformers import ( ImportError: cannot import name 'AutoModelWithLMHead' from 'transformers' (/usr/local/lib/python3.10/site-packages/transformers/__init__.py)

Container logs: