Upload src/pipeline.py with huggingface_hub

src/pipeline.py

@@ -28,7 +28,7 @@ def load_pipeline() -> Pipeline:
             path, 
             use_safetensors=False,
             local_files_only=True,
-            torch_dtype=torch.bfloat16)
+            torch_dtype=torch.bfloat16).to(memory_format=torch.channels_last)
     
     pipeline = FluxPipeline.from_pretrained(
         CHECKPOINT,
@@ -38,7 +38,7 @@ def load_pipeline() -> Pipeline:
         torch_dtype=torch.bfloat16,
     ).to("cuda")
 
-    pipeline.to(memory_format=torch.channels_last)
+#    pipeline.to(memory_format=torch.channels_last)
     for _ in range(4):
         pipeline("cat", num_inference_steps=4)