Skip to content

base_image_generation_pipeline_driver

BaseDiffusionImageGenerationPipelineDriver

Bases: ABC

Source code in griptape/drivers/image_generation_pipeline/base_image_generation_pipeline_driver.py 
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
@define
class BaseDiffusionImageGenerationPipelineDriver(ABC):
    @abstractmethod
    def prepare_pipeline(self, model: str, device: str | None) -> Any:
        pass

    @abstractmethod
    def make_image_param(self, image: Image | None) -> dict[str, Image] | None:
        pass

    @abstractmethod
    def make_additional_params(self, negative_prompts: list[str] | None, device: str | None) -> dict:
        pass

    @property
    @abstractmethod
    def output_image_dimensions(self) -> tuple[int, int]:
        pass

output_image_dimensions abstractmethod property

make_additional_params(negative_prompts, device) abstractmethod

Source code in griptape/drivers/image_generation_pipeline/base_image_generation_pipeline_driver.py 
22
23
24
@abstractmethod
def make_additional_params(self, negative_prompts: list[str] | None, device: str | None) -> dict:
    pass

make_image_param(image) abstractmethod

Source code in griptape/drivers/image_generation_pipeline/base_image_generation_pipeline_driver.py 
18
19
20
@abstractmethod
def make_image_param(self, image: Image | None) -> dict[str, Image] | None:
    pass

prepare_pipeline(model, device) abstractmethod

Source code in griptape/drivers/image_generation_pipeline/base_image_generation_pipeline_driver.py 
14
15
16
@abstractmethod
def prepare_pipeline(self, model: str, device: str | None) -> Any:
    pass