Keep VAE parameters out of inference mode when moving devices 44f12a2 verified topabaem commited on 1 day ago