if save can support ,safe_serialization=False

[NLPJCL]

The current save function cannot be set safe_serialization。but defalut safe_serialization=True。thank！

[tomaarsen]

Hello!

You're right, it would be nice if users could set safe_serialization so that they can save a pytorch_model.bin if they prefer.

• Tom Aarsen

[CKeibel]

Hi @tomaarsen !
If no one is working on this issue yet, could I work on this to get started with contributing?

[tomaarsen]

That would be great!
I think we would need to wrap this line with if isinstance(module, Transformer): and then pass the safe_serialization down to the Transformer, while all other modules still stay as module.save(model_path).

• Tom Aarsen

[CKeibel]

Great, thanks for the tip. I'll have a look at it.