|
@@ -1069,7 +1069,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
|
|
|
return samples
|
|
|
|
|
|
if self.latent_scale_mode is None:
|
|
|
- decoded_samples = torch.stack(decode_latent_batch(self.sd_model, samples, target_device=devices.cpu, check_for_nans=True))
|
|
|
+ decoded_samples = torch.stack(decode_latent_batch(self.sd_model, samples, target_device=devices.cpu, check_for_nans=True)).to(dtype=torch.float32)
|
|
|
else:
|
|
|
decoded_samples = None
|
|
|
|