Update src/pipeline.py

src/pipeline.py

@@ -53,7 +53,7 @@ def load_pipeline() -> Pipeline:
 
     model_id = "manbeast3b/flux-schnell-int8"
     transformer = FluxTransformer2DModel.from_pretrained(
-        cache_dir, subfolder="transformer", torch_dtype=torch.bfloat16,  # quantization_config=config,
+        cache_dir, subfolder="transformer", torch_dtype=torch.bfloat16, quantization_config=config
     )
     text_encoder_2 = T5EncoderModel.from_pretrained(
         model_id, subfolder="text_encoder_2", torch_dtype=torch.bfloat16