Update src/pipeline.py

src/pipeline.py

@@ -1,7 +1,7 @@
 from diffusers import FluxPipeline, AutoencoderKL, AutoencoderTiny
 from diffusers.image_processor import VaeImageProcessor
 from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
-
+from transformers import BitsAndBytesConfig
 from transformers import T5EncoderModel, T5TokenizerFast, CLIPTokenizer, CLIPTextModel
 import torch
 import torch._dynamo