Add back mega pipeline.

app.py

@@ -67,7 +67,8 @@ class DropOutput(Callback):
 from diffusers import EulerAncestralDiscreteScheduler
 scheduler = EulerAncestralDiscreteScheduler.from_config("runwayml/stable-diffusion-v1-5", subfolder="scheduler",
                                                         use_auth_token=AUTH_TOKEN)
-pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32,
+pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", custom_pipeline="stable_diffusion_mega",
+                                             torch_dtype=torch.float32,
                                              scheduler=scheduler, use_auth_token=AUTH_TOKEN)
 # pipe.enable_attention_slicing()
 # pipeline.to("cuda")