Setfit can't doing model prediction after training for MultiClass Classification with hf Trainer

[akkocz17]

When I train setfit for multiclass classification using huggingface trainer which I am using save_step arguments, I can not use the model.predict() method for getting predictions. I am getting the below not fitted error.
model.predict() only work when i am loading the model with save_pretrained() method, but i still can not use other saved steps models for prediction, there is only one useable model which saved with save_pretrained_method().

I did not face with the issue when train binary classification model using same setup.

[kgourgou]

Hello! Any chance you can provide some minimal code that replicates the issue? And also, which version of SetFit are you using?