Commit c09bc2c6 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

fix "clamp_scalar_cpu" not implemented for 'Half'

parent 4d9b0966
Loading
Loading
Loading
Loading
+1 −1
Original line number Original line Diff line number Diff line
@@ -1069,7 +1069,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
            return samples
            return samples


        if self.latent_scale_mode is None:
        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:
        else:
            decoded_samples = None
            decoded_samples = None