微调bge的时候报错，求助～

[passionate11]

报错如下 Traceback (most recent call last): File "/opt/tiger/ecom_Rag/FlagEmbedding/baai_general_embedding/finetune/run.py", line 222, in main() File "/opt/tiger/ecom_Rag/FlagEmbedding/baai_general_embedding/finetune/run.py", line 211, in main trainer.train() File "/usr/local/lib/python3.9/dist-packages/transformers/trainer.py", line 1885, in train return inner_training_loop( File "/usr/local/lib/python3.9/dist-packages/transformers/trainer.py", line 2147, in _inner_training_loop self.control = self.callback_handler.on_train_begin(args, self.state, self.control) File "/usr/local/lib/python3.9/dist-packages/transformers/trainer_callback.py", line 454, in on_train_begin return self.call_event("on_train_begin", args, state, control) File "/usr/local/lib/python3.9/dist-packages/transformers/trainer_callback.py", line 498, in call_event result = getattr(callback, event)( File "/usr/local/lib/python3.9/dist-packages/transformers/integrations/integration_utils.py", line 842, in on_train_begin self.setup(args, state, model, **kwargs) File "/usr/local/lib/python3.9/dist-packages/transformers/integrations/integration_utils.py", line 815, in setup model.save_pretrained(temp_dir) File "/usr/local/lib/python3.9/dist-packages/torch/nn/modules/module.py", line 1695, in getattr raise AttributeError(f"'{type(self).name}' object has no attribute '{name}'") AttributeError: 'BiEncoderModel' object has no attribute 'save_pretrained' 请问这是什么问题

[staoxiao]

This error is strange, and we don't use save_pretrained for BiEncoderModel. You can try to update the FlagEmbedding and transformers package.

[DreamInvoker]

I encountered the same problem when using the newest transformers 4.41.0.

When I downgrade it to 4.40.2., problem solved！ @passionate11 @staoxiao