wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-07 15:52:32 +08:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

first modification

Browse Source
This commit is contained in:
sylym 2023-03-29 21:47:24 +08:00
parent e65b693545
commit 5e75cbacd4
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -8,4 +8,3 @@ temp/
custom_nodes/ custom_nodes/
!custom_nodes/example_node.py.example !custom_nodes/example_node.py.example
extra_model_paths.yaml extra_model_paths.yaml
.idea/

2
comfy/samplers.py
View File

@ -221,8 +221,6 @@ class KSamplerX0Inpaint(torch.nn.Module):
def forward(self, x, sigma, uncond, cond, cond_scale, denoise_mask, cond_concat=None): def forward(self, x, sigma, uncond, cond, cond_scale, denoise_mask, cond_concat=None):
if denoise_mask is not None: if denoise_mask is not None:
latent_mask = 1. - denoise_mask 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 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) out = self.inner_model(x, sigma, cond=cond, uncond=uncond, cond_scale=cond_scale, cond_concat=cond_concat)
if denoise_mask is not None: if denoise_mask is not None: