Fixes race condition in training when VAE is unloaded
set_current_image can attempt to use the VAE when it is unloaded to the CPU while training
Showing
Please
register
or
sign in
to comment
set_current_image can attempt to use the VAE when it is unloaded to the CPU while training