We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
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
The text was updated successfully, but these errors were encountered:
No branches or pull requests
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
The text was updated successfully, but these errors were encountered: