Update app.py

app.py

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