AlekseyCalvin commited on
Commit
1fa6555
·
verified ·
1 Parent(s): 6a00986

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -1,6 +1,6 @@
1
  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
6
  from diffusers.loaders import FluxLoraLoaderMixin, FromSingleFileMixin, SD3LoraLoaderMixin
 
1
  import torch
2
  import numpy as np
3
+ from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast, CLIPTextModelWithProjection
4
  from diffusers import FlowMatchEulerDiscreteScheduler, AutoPipelineForImage2Image, FluxPipeline, FluxTransformer2DModel
5
  from diffusers import StableDiffusion3Pipeline, AutoencoderKL, DiffusionPipeline
6
  from diffusers.loaders import FluxLoraLoaderMixin, FromSingleFileMixin, SD3LoraLoaderMixin