Fix deprecated float16/fp16 variant loading through new `version` API. (#2)

end of training. runwayml/stable-diffusion-v1-5. vae from stabilityai/sd-vae-ft-mse. lr=5e-7. 4000 steps. batch size=2. mixed precision. trained text encoder. 1k reg images. 50 instance images.