modules/processing.py


1
2
3
4
5
6
7
8
9
10
11
12
13

class StableDiffusionProcessing:

    def __init__(self, init_imgs):
        self.init_images = init_imgs
        self.width = init_imgs[0].width
        self.height = init_imgs[0].height
        self.extra_generation_params = {}


class StableDiffusionProcessingImg2Img(StableDiffusionProcessing):

    def __init__(self, init_img):
        super().__init__(init_img)