瀏覽代碼

fix "clamp_scalar_cpu" not implemented for 'Half'

AUTOMATIC1111 2 年之前
父節點
當前提交
c09bc2c608
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      modules/processing.py

+ 1 - 1
modules/processing.py

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