Skip to content

Commit

Permalink
Backward compatibility patch for changes in the method signature of `…
Browse files Browse the repository at this point in the history
…InpaintModelConditioning`. (comfyanonymous#5825)

comfyanonymous#5813
  • Loading branch information
ltdrdata authored and tiangles committed Nov 29, 2024
1 parent 88b2c69 commit eceb1b9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nodes.py
Original file line number Diff line number Diff line change
Expand Up @@ -392,7 +392,7 @@ def INPUT_TYPES(s):

CATEGORY = "conditioning/inpaint"

def encode(self, positive, negative, pixels, vae, mask, noise_mask):
def encode(self, positive, negative, pixels, vae, mask, noise_mask=True):
x = (pixels.shape[1] // 8) * 8
y = (pixels.shape[2] // 8) * 8
mask = torch.nn.functional.interpolate(mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1])), size=(pixels.shape[1], pixels.shape[2]), mode="bilinear")
Expand Down

0 comments on commit eceb1b9

Please sign in to comment.