@@ -56,6 +56,7 @@ class CFGDenoiser(torch.nn.Module):
self.sampler=sampler
self.model_wrap=None
self.p=None
self.mask_before_denoising=False
@property
definner_model(self):
...
...
@@ -104,7 +105,7 @@ class CFGDenoiser(torch.nn.Module):
assertnotis_edit_modelorall(len(conds)==1forcondsinconds_list),"AND is not supported for InstructPix2Pix checkpoint (unless using Image CFG scale = 1.0)"