Skip to content

image_generation_model

__all__ = ['BaseImageGenerationModelDriver'] module-attribute

BaseImageGenerationModelDriver

Bases: SerializableMixin, ABC

Source code in griptape/drivers/image_generation_model/base_image_generation_model_driver.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
@define
class BaseImageGenerationModelDriver(SerializableMixin, ABC):
    @abstractmethod
    def get_generated_image(self, response: dict) -> bytes:
        pass

    @abstractmethod
    def text_to_image_request_parameters(
        self,
        prompts: list[str],
        image_width: int,
        image_height: int,
        negative_prompts: list[str] | None = None,
        seed: int | None = None,
    ) -> dict[str, Any]:
        pass

    @abstractmethod
    def image_variation_request_parameters(
        self,
        prompts: list[str],
        image: ImageArtifact,
        negative_prompts: list[str] | None = None,
        seed: int | None = None,
    ) -> dict[str, Any]:
        pass

    @abstractmethod
    def image_inpainting_request_parameters(
        self,
        prompts: list[str],
        image: ImageArtifact,
        mask: ImageArtifact,
        negative_prompts: list[str] | None = None,
        seed: int | None = None,
    ) -> dict[str, Any]:
        pass

    @abstractmethod
    def image_outpainting_request_parameters(
        self,
        prompts: list[str],
        image: ImageArtifact,
        mask: ImageArtifact,
        negative_prompts: list[str] | None = None,
        seed: int | None = None,
    ) -> dict[str, Any]:
        pass

get_generated_image(response) abstractmethod

Source code in griptape/drivers/image_generation_model/base_image_generation_model_driver.py 
16
17
18
@abstractmethod
def get_generated_image(self, response: dict) -> bytes:
    pass

image_inpainting_request_parameters(prompts, image, mask, negative_prompts=None, seed=None) abstractmethod

Source code in griptape/drivers/image_generation_model/base_image_generation_model_driver.py 
41
42
43
44
45
46
47
48
49
50
@abstractmethod
def image_inpainting_request_parameters(
    self,
    prompts: list[str],
    image: ImageArtifact,
    mask: ImageArtifact,
    negative_prompts: list[str] | None = None,
    seed: int | None = None,
) -> dict[str, Any]:
    pass

image_outpainting_request_parameters(prompts, image, mask, negative_prompts=None, seed=None) abstractmethod

Source code in griptape/drivers/image_generation_model/base_image_generation_model_driver.py 
52
53
54
55
56
57
58
59
60
61
@abstractmethod
def image_outpainting_request_parameters(
    self,
    prompts: list[str],
    image: ImageArtifact,
    mask: ImageArtifact,
    negative_prompts: list[str] | None = None,
    seed: int | None = None,
) -> dict[str, Any]:
    pass

image_variation_request_parameters(prompts, image, negative_prompts=None, seed=None) abstractmethod

Source code in griptape/drivers/image_generation_model/base_image_generation_model_driver.py 
31
32
33
34
35
36
37
38
39
@abstractmethod
def image_variation_request_parameters(
    self,
    prompts: list[str],
    image: ImageArtifact,
    negative_prompts: list[str] | None = None,
    seed: int | None = None,
) -> dict[str, Any]:
    pass

text_to_image_request_parameters(prompts, image_width, image_height, negative_prompts=None, seed=None) abstractmethod

Source code in griptape/drivers/image_generation_model/base_image_generation_model_driver.py 
20
21
22
23
24
25
26
27
28
29
@abstractmethod
def text_to_image_request_parameters(
    self,
    prompts: list[str],
    image_width: int,
    image_height: int,
    negative_prompts: list[str] | None = None,
    seed: int | None = None,
) -> dict[str, Any]:
    pass