image_generation_pipeline
Adapted from the Griptape AI Framework documentation.
__all__ = ['BaseDiffusionImageGenerationPipelineDriver']module-attribute
Bases:
ABCSource Code in griptape/drivers/image_generation_pipeline/base_image_generation_pipeline_driver.py
@define class BaseDiffusionImageGenerationPipelineDriver(ABC): @abstractmethod def prepare_pipeline(self, model: str, device: Optional[str]) -> Any: ... @abstractmethod def make_image_param(self, image: Optional[Image]) -> Optional[dict[str, Image]]: ... @abstractmethod def make_additional_params(self, negative_prompts: Optional[list[str]], device: Optional[str]) -> dict: ... @property @abstractmethod def output_image_dimensions(self) -> tuple[int, int]: ...
output_image_dimensionsabstractmethod property
make_additional_params(negative_prompts, device)abstractmethod
Source Code in griptape/drivers/image_generation_pipeline/base_image_generation_pipeline_driver.py
@abstractmethod def make_additional_params(self, negative_prompts: Optional[list[str]], device: Optional[str]) -> dict: ...
make_image_param(image)abstractmethod
Source Code in griptape/drivers/image_generation_pipeline/base_image_generation_pipeline_driver.py
@abstractmethod def make_image_param(self, image: Optional[Image]) -> Optional[dict[str, Image]]: ...
prepare_pipeline(model, device)abstractmethod
Source Code in griptape/drivers/image_generation_pipeline/base_image_generation_pipeline_driver.py
@abstractmethod def prepare_pipeline(self, model: str, device: Optional[str]) -> Any: ...
Could this page be better? Report a problem or suggest an addition!