Update src/pipeline.py

src/pipeline.py

@@ -21,15 +21,15 @@ torch._inductor.config.epilogue_fusion = False
 torch._inductor.config.coordinate_descent_check_all_directions = True
 os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
 
-CHECKPOINT = "jokerbit/flux.1-schnell-Robert-int8wo"
-REVISION = "5ef0012f11a863e5111ec56540302a023bc8587b"
+CHECKPOINT = "manbeast3b/Flux.1.schnell-quant2"
+REVISION = "44eb293715147878512da10bf3bc47cd14ec8c55"
 
 TinyVAE = "madebyollin/taef1"
 TinyVAE_REV = "2d552378e58c9c94201075708d7de4e1163b2689"
 
 
 def load_pipeline() -> Pipeline:
-    path = os.path.join(HF_HUB_CACHE, "models--jokerbit--flux.1-schnell-Robert-int8wo/snapshots/5ef0012f11a863e5111ec56540302a023bc8587b/transformer")
+    path = os.path.join(HF_HUB_CACHE, "models--manbeast3b--Flux.1.schnell-quant2/snapshots/44eb293715147878512da10bf3bc47cd14ec8c55/transformer")
     transformer = FluxTransformer2DModel.from_pretrained(
             path, 
             use_safetensors=False,