Update src/pipeline.py

src/pipeline.py

@@ -849,24 +849,24 @@ def load_pipeline() -> Pipeline:
     model_name = "manbeast3b/Flux.1.Schnell-full-quant1"
     revision = "e7ddf488a4ea8a3cba05db5b8d06e7e0feb826a2"
     
-    hub_model_dir = os.path.join(
-        HF_HUB_CACHE,
-        f"models--{model_name.replace('/', '--')}",
-        "snapshots",
-        revision,
-        "transformer"
-    )
-    transformer = FluxTransformer2DModel.from_pretrained(
-        hub_model_dir,
-        torch_dtype=torch.bfloat16,
-        use_safetensors=False
-    ).to(memory_format=torch.channels_last)
+    # hub_model_dir = os.path.join(
+    #     HF_HUB_CACHE,
+    #     f"models--{model_name.replace('/', '--')}",
+    #     "snapshots",
+    #     revision,
+    #     "transformer"
+    # )
+    # transformer = FluxTransformer2DModel.from_pretrained(
+    #     hub_model_dir,
+    #     torch_dtype=torch.bfloat16,
+    #     use_safetensors=False
+    # ).to(memory_format=torch.channels_last)
     
     pipeline = FluxPipeline.from_pretrained(
         ckpt_id,
         revision=ckpt_revision,
         # text_encoder_2=text_encoder_2,
-        transformer=transformer,
+        transformer=None,
         # vae=vae,
         torch_dtype=torch.bfloat16
     )