NameError: name 'Wav2Vec2BertModel' is not defined

[tianyashu2020]

/home/yunfeiyang/anaconda3/envs/xcodec2/lib/python3.9/site-packages/hydra/_internal/defaults_list.py:251: UserWarning: In 'default': Defaults list is missing _self_. See https://hydra.cc/docs/1.2/upgrades/1.0_to_1.1/default_composition_order for more information
warnings.warn(msg, UserWarning)
/home/yunfeiyang/anaconda3/envs/xcodec2/lib/python3.9/site-packages/torch/nn/utils/weight_norm.py:134: FutureWarning: torch.nn.utils.weight_norm is deprecated in favor of torch.nn.utils.parametrizations.weight_norm.
WeightNorm.apply(module, name, dim)
Error executing job with overrides: []
Traceback (most recent call last):
File "/home/yunfeiyang/xcodec2/./train.py", line 32, in train
lightning_module = CodecLightningModule(cfg)
File "/home/yunfeiyang/xcodec2/lightning_module.py", line 31, in init
self.construct_model()
File "/home/yunfeiyang/xcodec2/lightning_module.py", line 116, in construct_model
self.semantic_model = Wav2Vec2BertModel.from_pretrained(w2v_bert, output_hidden_states=True)
NameError: name 'Wav2Vec2BertModel' is not defined

[zhenye234]

sorry i miss this line from transformers import AutoFeatureExtractor, Wav2Vec2BertModel