Upload folder using huggingface_hub

src/pipeline.py

@@ -11,8 +11,7 @@ from torch import Generator
 from torchao.quantization import quantize_, int8_weight_only
 from transformers import T5EncoderModel, CLIPTextModel
 
-# torch.backends.cudnn.benchmark = True
-# torch.backends.cuda.matmul.allow_tf32 = True
+
 Pipeline: TypeAlias = FluxPipeline
 
 CHECKPOINT = "jokerbit/flux.1-schnell-Robert-int8wo"
@@ -53,7 +52,8 @@ def load_pipeline() -> Pipeline:
         torch_dtype=torch.bfloat16,
         use_safetensors=False,
     )
-
+    import pdb
+    pdb.set_trace()
     pipeline = FluxPipeline.from_pretrained(
         CHECKPOINT,
         revision=REVISION,