Update src/pipeline.py

src/pipeline.py

@@ -144,8 +144,8 @@ def load_pipeline() -> Pipeline:
     pipeline.text_encoder.to(memory_format=torch.channels_last)
     pipeline.text_encoder_2.to(memory_format=torch.channels_last)
     pipeline.transformer.to(memory_format=torch.channels_last)
-    pipeline._exclude_from_cpu_offload = ["vae"]
-    pipeline.enable_sequential_cpu_offload()
+    # pipeline._exclude_from_cpu_offload = ["vae"]
+    # pipeline.enable_sequential_cpu_offload()
     for _ in range(2):
         pipeline(prompt="onomancy, aftergo, spirantic, Platyhelmia, modificator, drupaceous, jobbernowl, hereness", width=1024, height=1024, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256)