Model loading. Isn't this a bad idea?

[ootsuka-repos]

The file name of train's save in progress is not good looking.
I think you should fix the save name because it is hard-coded in init.py. Also, there is no config.json, so I want the save. It's a pain to download it from https://huggingface.co/declare-lab/TangoFlux and fix it by hand when you change parameters!