Remove the optimiser before uploading the model to huggingface

openclimatefix / PVNet

PVnet main repo

MIT License

15 stars 3 forks source link

Remove the optimiser before uploading the model to huggingface #225

Open dfulu opened 2 weeks ago

dfulu commented 2 weeks ago

Due to momentum, the optimiser can take up a lot disk space compared to the model itself. We currently upload the full model, including the optimiser to huggingface and download that to the app. We should remove the optimiser before uploading to save on bandwidth and memory