Save the model into a .h5 model

Hi there, I have fine-tuned the ViTForImageClassification and now I save into a folder containing:

  • config.json
  • tf_model.h5

but the point is that the .h5 doesn’t contain the model but just the weights.

There is a way to get the whole model in one .h5 model? It’s possible in keras so I think that there is a solution for it!