fix "clamp_scalar_cpu" not implemented for 'Half'

This commit is contained in:
AUTOMATIC1111 2023-07-31 13:20:26 +03:00
parent 4d9b096663
commit c09bc2c608

View File

@ -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