diff options
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py new file mode 100644 index 0000000..649c3a1 --- /dev/null +++ b/modules/processing.py @@ -0,0 +1,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)
|