Skip to content

Commit 5c94937

Browse files
authored
Update pipeline_flux_img2img.py (#9928)
* Update pipeline_flux_img2img.py Added FromSingleFileMixin to this pipeline loader like the other FLUX pipelines. * Update pipeline_flux_img2img.py typo * modified: src/diffusers/pipelines/flux/pipeline_flux_img2img.py
1 parent d74483c commit 5c94937

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/flux/pipeline_flux_img2img.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
2121

2222
from ...image_processor import PipelineImageInput, VaeImageProcessor
23-
from ...loaders import FluxLoraLoaderMixin, TextualInversionLoaderMixin
23+
from ...loaders import FluxLoraLoaderMixin, FromSingleFileMixin, TextualInversionLoaderMixin
2424
from ...models.autoencoders import AutoencoderKL
2525
from ...models.transformers import FluxTransformer2DModel
2626
from ...schedulers import FlowMatchEulerDiscreteScheduler
@@ -159,7 +159,7 @@ def retrieve_timesteps(
159159
return timesteps, num_inference_steps
160160

161161

162-
class FluxImg2ImgPipeline(DiffusionPipeline, FluxLoraLoaderMixin):
162+
class FluxImg2ImgPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFileMixin):
163163
r"""
164164
The Flux pipeline for image inpainting.
165165

0 commit comments

Comments
 (0)