ssit
c2e7ca4043
- Removed checkpoint related inputs for easier testing. - Added modules that mimic A1111 modules
31 lines
770 B
Python
31 lines
770 B
Python
# Patched classes to adapt from A111 webui for ComfyUI
|
|
|
|
class StableDiffusionProcessing:
|
|
seed = 0
|
|
extra_generation_params = {}
|
|
|
|
def __init__(self, init_img):
|
|
self.init_images = [init_img]
|
|
|
|
|
|
class Processed:
|
|
|
|
def __init__(self, p: StableDiffusionProcessing, images: list, seed: int, info: str):
|
|
self.images = images
|
|
self.seed = seed
|
|
self.info = info
|
|
|
|
def infotext(self, p: StableDiffusionProcessing, index):
|
|
return None
|
|
|
|
|
|
def fix_seed(p: StableDiffusionProcessing):
|
|
pass
|
|
|
|
|
|
def process_images(p: StableDiffusionProcessing) -> Processed:
|
|
# Where the main image generation happens in A1111
|
|
# Return original images for now
|
|
processed = Processed(p, p.init_images, p.seed, None)
|
|
return processed
|