Update app.py

app.py

@@ -8,6 +8,7 @@ accelerator = Accelerator(cpu=True)
 
 pipe = accelerator.prepare(StableDiffusionXLPipeline.from_pretrained("segmind/Segmind-Vega", torch_dtype=torch.float32, use_safetensors=True, variant=None,safety_checker=False))
 pipe.scheduler = accelerator.prepare(EulerDiscreteScheduler.from_config(pipe.scheduler.config))
+pipe.unet.to(memory_format=torch.channels_last)
 pipe = accelerator.prepare(pipe.to("cpu"))
 apol=[]
 def plex(prompt,neg_prompt,stips,nut):