cannot import name 'PreTrainedModel' from 'transformers'

[missTL]

Installed transformers-4.57.0 but encountered an error. However, it contains PreTrainedModel, yet the error still persists:

Traceback (most recent call last):
File "/checkpoint/binary/train_package/playground/benchmarks/discrete_vla_pretrain.py", line 39, in
from dexbotic.model.discrete_vla.discrete_vla_arch import DiscreteVLAForCausalLM
File "/checkpoint/binary/train_package/dexbotic/model/discrete_vla/discrete_vla_arch.py", line 5, in
from dexbotic.model.dexbotic_arch import (ActionOutputForCausalLM,
File "/checkpoint/binary/train_package/dexbotic/model/dexbotic_arch.py", line 8, in
from transformers import (AutoConfig, AutoModel, PretrainedConfig,
ImportError: cannot import name 'PreTrainedModel' from 'transformers' (/root/.local/lib/python3.10/site-packages/transformers/init.py)

[Rocketknight1]

Strange, can you try the latest version on main? I suspect this might be an environment issue.