提交 02038036 编写于 作者: A AUTOMATIC1111

make it so that VAE NaNs autodetection also works during first pass of hires fix

上级 cc53db66
......@@ -1069,7 +1069,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
return samples
if self.latent_scale_mode is None:
decoded_samples = decode_first_stage(self.sd_model, samples)
decoded_samples = torch.stack(decode_latent_batch(self.sd_model, samples, target_device=devices.cpu, check_for_nans=True))
else:
decoded_samples = None
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册