diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-16 12:28:50 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-16 12:28:50 +0300 |
| commit | 570f42afd122405116b39b880cdb5163fd5ca3e2 (patch) | |
| tree | bfe5dce49561009da46bcb19e7c6aef4b04d12e2 /modules/processing.py | |
| parent | 0198eaec455157a7dc1c950708d1ec95bcf4629c (diff) | |
possible fix for FP16 VAE failing in img2img SDXL
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/processing.py b/modules/processing.py index e7b10808..6567b3cf 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -1303,7 +1303,7 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing): image = torch.from_numpy(batch_images)
image = 2. * image - 1.
- image = image.to(shared.device)
+ image = image.to(shared.device, dtype=devices.dtype_vae)
self.init_latent = self.sd_model.get_first_stage_encoding(self.sd_model.encode_first_stage(image))
|