Upload src/pipeline.py with huggingface_hub

src/pipeline.py

@@ -29,17 +29,11 @@ def load_pipeline() -> Pipeline:
             use_safetensors=False,
             local_files_only=True,
             torch_dtype=torch.bfloat16).to(memory_format=torch.channels_last)
-    vae = AutoencoderTiny.from_pretrained(
-            TinyVAE,
-            revision=TinyVAE_REV,
-            local_files_only=True,
-            torch_dtype=torch.bfloat16
-            )    
+
     pipeline = FluxPipeline.from_pretrained(
         CHECKPOINT,
         revision=REVISION,
         transformer=transformer,
-        vae=vae,
         local_files_only=True,
         torch_dtype=torch.bfloat16,
     )