Upload src/pipeline.py with huggingface_hub

src/pipeline.py

@@ -44,15 +44,14 @@ def load_pipeline() -> Pipeline:
     quantization_config = BitsAndBytesConfig(
                 load_in_8bit=True, llm_int8_threshold=10.,
                 )
+    path = os.path.join(HF_HUB_CACHE, "models--black-forest-labs--FLUX.1-schnell/snapshots/741f7c3ce8b383c54771c7003378a50191e9efe9/transformer")
     transformer = FluxTransformer2DModel.from_pretrained(
-        CHECKPOINT,
-        revision=REVISION,
-        subfolder="transformer",
+        path,
         torch_dtype=torch.bfloat16,
         quantization_config=quantization_config,
-        local_files_only=True,
-        use_safetensors=True,
-    )
+#        local_files_only=True,
+#        use_safetensors=True,
+        )
 
 
     pipeline = FluxPipeline.from_pretrained(