Update app.py

app.py

@@ -9,10 +9,10 @@ from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
 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 = accelerator.prepare(WuerstchenPriorPipeline.from_pretrained("warp-ai/wuerstchen-prior", torch_dtype=torch.float32, use_safetensors=True, safety_cheker=None))
 ###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 = accelerator.prepare(WuerstchenDecoderPipeline.from_pretrained("warp-ai/wuerstchen", torch_dtype=torch.float32, use_safetensors=True, safety_checker=None))
 ###warp.scheduler = DDPMWuerstchenScheduler.from_config(warp.scheduler.config)
 warp = warp.to("cpu")
 generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4876364))