Update app.py

app.py

@@ -115,11 +115,11 @@ sdulers =[
 def plex(mput, prompt, neg_prompt, stips, modal_id, dula, blip, blop):
     dula=dula ## shedulers todo
     pope = accelerator.prepare(StableDiffusionPipeline.from_pretrained(modal_id, use_safetensors=False, safety_checker=None,torch_dtype=torch.float32))
-    pope.to("cpu")
+    pope = accelerator.prepare(pope.to("cpu"))
     pipe = accelerator.prepare(StableDiffusionControlNetPipeline.from_pretrained(modal_id, use_safetensors=False,controlnet=controlnet, safety_checker=None,torch_dtype=torch.float32))
-    pipe.to("cpu")
+    pipe = accelerator.prepare(pipe.to("cpu"))
     pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
-    generator = torch.Generator(device="cpu").manual_seed(random.randint(0, MAX_SEED))
+    generator = torch.Generator("cpu").manual_seed(random.randint(0, MAX_SEED))
     
     tilage = pope(prompt,num_inference_steps=5).images[0]
     cannmage = tilage