Model loading. Isn't this a bad idea? #22
Description
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!
