Update src/pipeline.py

src/pipeline.py

@@ -49,13 +49,13 @@ def load_pipeline() -> Pipeline:
 
     model_id = "manbeast3b/flux-schnell-int8"
     transformer = FluxTransformer2DModel.from_pretrained(
-        model_id, subfolder="transformer", quantization_config=config, torch_dtype=torch.bfloat16
+        model_id, subfolder="transformer", torch_dtype=torch.bfloat16, local_files_only=True # quantization_config=config,
     )
     text_encoder_2 = T5EncoderModel.from_pretrained(
-        model_id, subfolder="text_encoder_2", quantization_config=config, torch_dtype=torch.bfloat16
+        model_id, subfolder="text_encoder_2", torch_dtype=torch.bfloat16
     )
     text_encoder = CLIPTextModel.from_pretrained(
-        model_id, subfolder="text_encoder", quantization_config=config, torch_dtype=torch.bfloat16
+        model_id, subfolder="text_encoder",torch_dtype=torch.bfloat16
     )
     # vae=AutoencoderKL.from_pretrained(ckpt_id, subfolder="vae", torch_dtype=dtype)
     pipeline = DiffusionPipeline.from_pretrained(