Open pratikchhapolika opened 6 months ago
I have trained a model using latest transformers package 4.37.2 and it saves the model as model.safetensors.
transformers
4.37.2
model.safetensors
I am using from fast_bert.prediction import BertClassificationPredictor to load the model in production. But it's giving the error as
from fast_bert.prediction import BertClassificationPredictor
OSError: Error no file named ['pytorch_model.bin', 'tf_model.h5', 'model.ckpt.index'] found in directory or `from_tf` set to False
Does BertClassificationPredictor allow us to load model.safetensors. If yes, how can we do this?
BertClassificationPredictor
predictor = BertClassificationPredictor(model_path=self.model_path, label_path=self.label_path, multi_label=False, model_type='roberta', do_lower_case=True)
@kaushaltrivedi
I have trained a model using latest
transformers
package4.37.2
and it saves the model asmodel.safetensors
.I am using
from fast_bert.prediction import BertClassificationPredictor
to load the model in production. But it's giving the error asDoes
BertClassificationPredictor
allow us to loadmodel.safetensors
. If yes, how can we do this?@kaushaltrivedi