Skip to content

Base multi model image generation driver

BaseMultiModelImageGenerationDriver

Bases: BaseImageGenerationDriver, ABC

Image Generation Driver for platforms like Amazon Bedrock that host many LLM models.

Instances of this Image Generation Driver require a Image Generation Model Driver which is used to structure the image generation request in the format required by the model and to process the output.

Attributes:

Name Type Description
image_generation_model_driver BaseImageGenerationModelDriver

Image Model Driver to use.

Source code in griptape/drivers/image_generation/base_multi_model_image_generation_driver.py
@define
class BaseMultiModelImageGenerationDriver(BaseImageGenerationDriver, ABC):
    """Image Generation Driver for platforms like Amazon Bedrock that host many LLM models.

    Instances of this Image Generation Driver require a Image Generation Model Driver which is used to structure the
    image generation request in the format required by the model and to process the output.

    Attributes:
        image_generation_model_driver: Image Model Driver to use.
    """

    image_generation_model_driver: BaseImageGenerationModelDriver = field(kw_only=True, metadata={"serializable": True})

image_generation_model_driver: BaseImageGenerationModelDriver = field(kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute