search

aws-neuron / transformers-neuronx

Apache License 2.0
100 stars 29 forks source link

Turn off safe_serialization from save_split so that save_function is called #56

Open jitto opened 1 year ago

jitto commented 1 year ago

Initial fix for #55 - so that save_split can work again.

Transformer v4.35.0 broke the save_split and this change will have the same function as transformer v4.34.1