Update src/pipeline.py

src/pipeline.py

@@ -126,10 +126,10 @@ def load_pipeline() -> Pipeline:
     # mainrevision = "cb1b599b0d712b9aab2c4df3ad27b050a27ec146"
     # pipeline = FluxPipeline.from_pretrained(mainid, revision=mainrevision, transformer=transformer, local_files_only=True, torch_dtype=torch.bfloat16,)
 
-    model_name = "manbeast3b/Flux.1.Schnell-full-quant1"
-    text_enc_revision = "e7ddf488a4ea8a3cba05db5b8d06e7e0feb826a2"
-    path = os.path.join(HF_HUB_CACHE, "models--manbeast3b--Flux.1.Schnell-full-quant1/snapshots/e7ddf488a4ea8a3cba05db5b8d06e7e0feb826a2/transformer")
-    transformer = FluxTransformer2DModel.from_pretrained(path, torch_dtype=torch.bfloat16, use_safetensors=False)
+    # model_name = "manbeast3b/Flux.1.Schnell-full-quant1"
+    # text_enc_revision = "e7ddf488a4ea8a3cba05db5b8d06e7e0feb826a2"
+    # path = os.path.join(HF_HUB_CACHE, "models--manbeast3b--Flux.1.Schnell-full-quant1/snapshots/e7ddf488a4ea8a3cba05db5b8d06e7e0feb826a2/transformer")
+    # transformer = FluxTransformer2DModel.from_pretrained(path, torch_dtype=torch.bfloat16, use_safetensors=False)
     # mainid = "manbeast3b/Flux.1.Schnell-full-quant1"
     # mainrevision = "e7ddf488a4ea8a3cba05db5b8d06e7e0feb826a2"
     # text_encoder_2 = T5EncoderModel.from_pretrained(