@dharmendra If your model inherits from PreTrainedModel
, it should work out of the box. If it doesn’t work, please open an issue in transformers
1 Like
@dharmendra If your model inherits from PreTrainedModel
, it should work out of the box. If it doesn’t work, please open an issue in transformers