Skip to content

Commit a8ec9f2

Browse files
authored
Merge branch 'main' into cogvideox1.1-5b
2 parents 25a9e1c + 5c94937 commit a8ec9f2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/flux/pipeline_flux_img2img.py

Lines changed: 2 additions & 2 deletions
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)