earth2studio
earth2studio copied to clipboard
🚀[FEA]: HENS Recipe Updates
Is this a new feature, an improvement, or a change to existing functionality?
Improvement
How would you describe the priority of this feature request
Medium
Please provide a clear description of problem you would like to solve.
Tracking a few items that can be improved with the HENS recipe:
- [ ] Support the huggingface pull to selectively pull checkpoints as needed
- [ ] Make dedicated HENS model wrapper in E2S (to discuss)
- [ ] Remove manual threading for IO and improve IO in base package
- [x] Clean up the configs more
- [ ] Work on eliminating ambiguous dictionary objects when possible
- [ ] Update Earth2Studio dep to release
- [x] Add testing command in make file (probably notebook is sufficient? Maybe reproduce sample?)