Update app.py

app.py

@@ -7,16 +7,16 @@ from diffusers.utils import load_image
 from diffusers import DiffusionPipeline, DDPMScheduler
 
 from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
-from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS, DDPMWuerstchenScheduler
+from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
 
 
 
 accelerator = Accelerator(cpu=True)
 warp_prior = accelerator.prepare(WuerstchenPriorPipeline.from_pretrained("warp-ai/wuerstchen-prior", torch_dtype=torch.bfloat16, use_safetensors=True, safety_cheker=None))
-warp_prior.scheduler = DDPMWuerstchenScheduler.from_config(warp_prior.scheduler.config)
+###warp_prior.scheduler = DDPMWuerstchenScheduler.from_config(warp_prior.scheduler.config)
 warp_prior = warp_prior.to("cpu")
 warp = accelerator.prepare(WuerstchenDecoderPipeline.from_pretrained("warp-ai/wuerstchen", torch_dtype=torch.bfloat16, use_safetensors=True, safety_checker=None))
-warp.scheduler = DDPMWuerstchenScheduler.from_config(warp.scheduler.config)
+###warp.scheduler = DDPMWuerstchenScheduler.from_config(warp.scheduler.config)
 warp = warp.to("cpu")
 generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4876364))