Update app.py

app.py

@@ -21,11 +21,11 @@ from tqdm import tqdm
 import supervision as sv
 
 accelerator = Accelerator(cpu=True)
-controlnet = accelerator.prepare(ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.bfloat16))
+controlnet = accelerator.prepare(ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float32))
 pipe = accelerator.prepare(StableDiffusionControlNetImg2ImgPipeline.from_pretrained("stabilityai/sdxl-turbo", controlnet=controlnet, torch_dtype=torch.bfloat16, use_safetensors=True, variant="fp16", safety_checker=None))
 pipe.unet.to(memory_format=torch.channels_last)
 pipe.scheduler = accelerator.prepare(EulerDiscreteScheduler.from_config(pipe.scheduler.config))
-pipe = pipe.to("cpu")
+pipe = accelerator.prepare(pipe.to("cpu"))
 
 def plex(fpath, text, neg_prompt, one, two, three, four, five):
     gc.collect()