Update src/pipeline.py

src/pipeline.py

@@ -30,6 +30,8 @@ def load_pipeline() -> Pipeline:
         text_encoder_2=text_encoder_2,
         torch_dtype=dtype,
     )#.to("cuda")
+    pipeline.transformer.to(memory_format=torch.channels_last)
+    pipeline.text_encoder.to(memory_format=torch.channels_last)
 
     # Optimize after moving to GPU
     pipeline.vae = torch.compile(pipeline.vae) # compile after moving to device