Diffusion-API / diffusion_webui /diffusion_models /base_controlnet_pipeline.py
kadirnar's picture
Upload 15 files
cab0202
raw
history blame
844 Bytes
class ControlnetPipeline:
def __init__(self):
self.pipe = None
def load_model(self, stable_model_path: str, controlnet_model_path: str):
raise NotImplementedError()
def load_image(self, image_path: str):
raise NotImplementedError()
def controlnet_preprocces(self, read_image: str):
raise NotImplementedError()
def generate_image(
self,
image_path: str,
stable_model_path: str,
controlnet_model_path: str,
prompt: str,
negative_prompt: str,
num_images_per_prompt: int,
guidance_scale: int,
num_inference_step: int,
controlnet_conditioning_scale: int,
scheduler: str,
seed_generator: int,
):
raise NotImplementedError()
def web_interface():
raise NotImplementedError()