option to use xattention optimizations when training
option to unload vae when training