AlekseyCalvin commited on
Commit
b3b6a55
1 Parent(s): 5164408

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +12 -2
pipeline.py CHANGED
@@ -1,7 +1,17 @@
1
  import torch
2
  import numpy as np
3
- from diffusers import FlowMatchEulerDiscreteScheduler, AutoPipelineForImage2Image, FluxPipeline
4
- from diffusers import StableDiffusion3Pipeline, AutoencoderKL
 
 
 
 
 
 
 
 
 
 
5
  from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
6
  from typing import Any, Callable, Dict, List, Optional, Union
7
  from PIL import Image
 
1
  import torch
2
  import numpy as np
3
+ from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
4
+ from diffusers import FlowMatchEulerDiscreteScheduler, AutoPipelineForImage2Image, FluxPipeline, FluxTransformer2DModel, VaeImageProcessor
5
+ from diffusers import StableDiffusion3Pipeline, AutoencoderKL, DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFileMixin, SD3LoraLoaderMixin
6
+ from diffusers.utils import (
7
+ USE_PEFT_BACKEND,
8
+ is_torch_xla_available,
9
+ logging,
10
+ replace_example_docstring,
11
+ scale_lora_layers,
12
+ unscale_lora_layers,
13
+ )
14
+ from diffusers.utils.torch_utils import randn_tensor
15
  from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
16
  from typing import Any, Callable, Dict, List, Optional, Union
17
  from PIL import Image