Save the model as pkl or pt
#5
by adityakad - opened
Hi Tom, I want to deploy the model as AWS SageMaker Endpoint but I don't want to use the model from the Hugging Face Hub.
I get an error when I try to save the model as a pkl file or a pt file. Is there a way to do that?
adityakad changed discussion title from Save the model file to Save the model as pkl or pt
I've never used AWS SageMaker. But the model is already in Pickle .bin format, so are you sure you need to convert it?
Yes, the way AWS SageMaker works in my company, it only accepts models in .pkl or .pt or .th formats. Would that be possible?