Update app.py

app.py

@@ -11,6 +11,7 @@ from accelerate import Accelerator
 accelerator = Accelerator()
 apol=[]
 pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=None))
+pipe.unet.to(memory_format=torch.channels_last)
 pipe = accelerator.prepare(pipe.to("cpu"))
 def plex(prompt,neg_prompt):
     nm = random.randint(1, 4836928)