Update src/pipeline.py

src/pipeline.py

@@ -31,7 +31,7 @@ def load_pipeline() -> Pipeline:
         torch_dtype=dtype,
     )#.to("cuda")
     pipeline.transformer.to(memory_format=torch.channels_last)
-    pipeline.text_encoder.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