diff --git a/ldm/invoke/generator/omnibus.py b/ldm/invoke/generator/omnibus.py index 35c4a62d66..277e897021 100644 --- a/ldm/invoke/generator/omnibus.py +++ b/ldm/invoke/generator/omnibus.py @@ -12,6 +12,8 @@ from ldm.invoke.generator.txt2img import Txt2Img 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,