Add basic implementation for AuraFlowImg2ImgPipeline

Author: AstraliteHeart

src/diffusers/__init__.py

@@ -346,6 +346,7 @@
             "AudioLDM2UNet2DConditionModel",
             "AudioLDMPipeline",
             "AuraFlowPipeline",
+            "AuraFlowImg2ImgPipeline",
             "BlipDiffusionControlNetPipeline",
             "BlipDiffusionPipeline",
             "CLIPImageProjection",

src/diffusers/pipelines/__init__.py

@@ -309,7 +309,7 @@
             "StableDiffusionLDM3DPipeline",
         ]
     )
-    _import_structure["aura_flow"] = ["AuraFlowPipeline"]
+    _import_structure["aura_flow"] = ["AuraFlowPipeline", "AuraFlowImg2ImgPipeline"]
     _import_structure["stable_diffusion_3"] = [
         "StableDiffusion3Pipeline",
         "StableDiffusion3Img2ImgPipeline",
@@ -515,7 +515,7 @@
             AudioLDM2ProjectionModel,
             AudioLDM2UNet2DConditionModel,
         )
-        from .aura_flow import AuraFlowPipeline
+        from .aura_flow import AuraFlowPipeline, AuraFlowImg2ImgPipeline
         from .blip_diffusion import BlipDiffusionPipeline
         from .cogvideo import (
             CogVideoXFunControlPipeline,

src/diffusers/pipelines/aura_flow/__init__.py

@@ -23,6 +23,7 @@
     _dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
 else:
     _import_structure["pipeline_aura_flow"] = ["AuraFlowPipeline"]
+    _import_structure["pipeline_aura_flow_img2img"] = ["AuraFlowImg2ImgPipeline"]
 
 if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
     try:
@@ -33,6 +34,7 @@
         from ...utils.dummy_torch_and_transformers_objects import *
     else:
         from .pipeline_aura_flow import AuraFlowPipeline
+        from .pipeline_aura_flow_img2img import AuraFlowImg2ImgPipeline
 
 else:
     import sys