diff --git a/.gitignore b/.gitignore index e1eaa55f3..ebcf36ddc 100644 --- a/.gitignore +++ b/.gitignore @@ -7,5 +7,4 @@ models/ temp/ custom_nodes/ !custom_nodes/example_node.py.example -extra_model_paths.yaml -.idea/ \ No newline at end of file +extra_model_paths.yaml \ No newline at end of file diff --git a/comfy/samplers.py b/comfy/samplers.py index 4da5d523a..9655cc35a 100644 --- a/comfy/samplers.py +++ b/comfy/samplers.py @@ -221,8 +221,6 @@ class KSamplerX0Inpaint(torch.nn.Module): def forward(self, x, sigma, uncond, cond, cond_scale, denoise_mask, cond_concat=None): if denoise_mask is not None: latent_mask = 1. - denoise_mask - if not torch.all(torch.eq(sigma, sigma[0])): - raise ValueError('sigma must be the same for all timesteps when using denoise_mask') x = x * denoise_mask + (self.latent_image + self.noise * sigma[0]) * latent_mask out = self.inner_model(x, sigma, cond=cond, uncond=uncond, cond_scale=cond_scale, cond_concat=cond_concat) if denoise_mask is not None: