diff --git a/ldm/invoke/generator/omnibus.py b/ldm/invoke/generator/omnibus.py index 35c4a62d664..277e8970210 100644 --- a/ldm/invoke/generator/omnibus.py +++ b/ldm/invoke/generator/omnibus.py @@ -12,6 +12,8 @@ class Omnibus(Img2Img,Txt2Img): def __init__(self, model, precision): super().__init__(model, precision) + self.pil_mask = None + self.pil_image = None def get_make_image( self,