Skip to content

Add FLUX image-to-image to AutoPipelineForImage2Image  #9492

Description

@apolinario

Is your feature request related to a problem? Please describe.
Currently FluxImg2ImgPipeline is not mapped to AutoPipelineForImage2Image

Describe the solution you'd like.
A clear and concise description of what you want to happen.

Additional context.
Here's the error

Traceback (most recent call last):
  File "/home/user/app/app.py", line 28, in <module>
    pipe_i2i = AutoPipelineForImage2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype).to(device)
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
    return fn(*args, **kwargs)
  File "/usr/local/lib/python3.10/site-packages/diffusers/pipelines/auto_pipeline.py", line 670, in from_pretrained
    image_2_image_cls = _get_task_class(AUTO_IMAGE2IMAGE_PIPELINES_MAPPING, orig_class_name)
  File "/usr/local/lib/python3.10/site-packages/diffusers/pipelines/auto_pipeline.py", line 214, in _get_task_class
    raise ValueError(f"AutoPipeline can't find a pipeline linked to {pipeline_class_name} for {model_name}")
ValueError: AutoPipeline can't find a pipeline linked to FluxPipeline for flux

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions