File size: 844 Bytes
95a9559 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
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()
|