Export new drivers

griptape-ai · Jul 25, 2024 · 71461bf · 71461bf
1 parent 413bbb4
commit 71461bf
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/griptape/drivers/__init__.py b/griptape/drivers/__init__.py
@@ -53,6 +53,18 @@
     BedrockStableDiffusionImageGenerationModelDriver,
 )
 from .image_generation_model.bedrock_titan_image_generation_model_driver import BedrockTitanImageGenerationModelDriver
+from .image_generation_model.stable_diffusion_3_pipeline_image_generation_model_driver import (
+    StableDiffusion3PipelineImageGenerationModelDriver,
+)
+from .image_generation_model.base_diffusion_pipeline_image_generation_model_driver import (
+    BaseDiffusionPipelineImageGenerationModelDriver,
+)
+from .image_generation_model.stable_diffusion_3_img_2_img_pipeline_image_generation_model_driver import (
+    StableDiffusion3Img2ImgPipelineImageGenerationModelDriver,
+)
+from .image_generation_model.stable_diffusion_3_controlnet_pipeline_image_generation_model_driver import (
+    StableDiffusion3ControlNetPipelineImageGenerationModelDriver,
+)
 
 from .image_generation.base_image_generation_driver import BaseImageGenerationDriver
 from .image_generation.base_multi_model_image_generation_driver import BaseMultiModelImageGenerationDriver
@@ -61,6 +73,9 @@
 from .image_generation.amazon_bedrock_image_generation_driver import AmazonBedrockImageGenerationDriver
 from .image_generation.azure_openai_image_generation_driver import AzureOpenAiImageGenerationDriver
 from .image_generation.dummy_image_generation_driver import DummyImageGenerationDriver
+from .image_generation.huggingface_diffusion_pipeline_image_generation_driver import (
+    HuggingFaceDiffusionPipelineImageGenerationDriver,
+)
 
 from .image_query_model.base_image_query_model_driver import BaseImageQueryModelDriver
 from .image_query_model.bedrock_claude_image_query_model_driver import BedrockClaudeImageQueryModelDriver
@@ -164,13 +179,18 @@
     "BaseImageGenerationModelDriver",
     "BedrockStableDiffusionImageGenerationModelDriver",
     "BedrockTitanImageGenerationModelDriver",
+    "BaseDiffusionPipelineImageGenerationModelDriver",
+    "StableDiffusion3PipelineImageGenerationModelDriver",
+    "StableDiffusion3Img2ImgPipelineImageGenerationModelDriver",
+    "StableDiffusion3ControlNetPipelineImageGenerationModelDriver",
     "BaseImageGenerationDriver",
     "BaseMultiModelImageGenerationDriver",
     "OpenAiImageGenerationDriver",
     "LeonardoImageGenerationDriver",
     "AmazonBedrockImageGenerationDriver",
     "AzureOpenAiImageGenerationDriver",
     "DummyImageGenerationDriver",
+    "HuggingFaceDiffusionPipelineImageGenerationDriver",
     "BaseImageQueryModelDriver",
     "BedrockClaudeImageQueryModelDriver",
     "BaseImageQueryDriver",