Right now we have to load in:
- The TAN checkpoint
- Config file
- TFs
- dims
(4) should be merged into (2); (3) should also be merged into (2) (see e.g. scuba repo); and then everything should be put into one folder so everything can be auto-loaded with one path