AlekseyCalvin commited on
Commit
6e146d7
1 Parent(s): 1a0dfbb

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -1
pipeline.py CHANGED
@@ -2,7 +2,8 @@ import torch
2
  import numpy as np
3
  from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
4
  from diffusers import FlowMatchEulerDiscreteScheduler, AutoPipelineForImage2Image, FluxPipeline, FluxTransformer2DModel
5
- from diffusers import StableDiffusion3Pipeline, AutoencoderKL, DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFileMixin, SD3LoraLoaderMixin
 
6
  from diffusers.utils import (
7
  USE_PEFT_BACKEND,
8
  is_torch_xla_available,
 
2
  import numpy as np
3
  from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
4
  from diffusers import FlowMatchEulerDiscreteScheduler, AutoPipelineForImage2Image, FluxPipeline, FluxTransformer2DModel
5
+ from diffusers import StableDiffusion3Pipeline, AutoencoderKL, DiffusionPipeline,
6
+ from diffusers.loaders import FluxLoraLoaderMixin, FromSingleFileMixin, SD3LoraLoaderMixin
7
  from diffusers.utils import (
8
  USE_PEFT_BACKEND,
9
  is_torch_xla_available,