set_current_image can attempt to use the VAE when it is unloaded to the CPU while training
Attach a file by drag & drop or click to upload