Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ accelerator = Accelerator(cpu=True)
|
|
13 |
warp_prior = accelerator.prepare(DiffusionPipeline.from_pretrained("warp-ai/wuerstchen-prior", torch_dtype=torch.bfloat16, use_safetensors=True, safety_cheker=None))
|
14 |
warp_prior.scheduler = DDPMScheduler.from_config(warp_prior.scheduler.config)
|
15 |
warp_prior = warp_prior.to("cpu")
|
16 |
-
warp = accelerator.prepare(DiffusionPipeline.from_pretrained("warp-ai/wuerstchen", torch_dtype=torch.bfloat16
|
17 |
warp.scheduler = DDPMScheduler.from_config(warp.scheduler.config)
|
18 |
warp = warp.to("cpu")
|
19 |
generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4876364))
|
|
|
13 |
warp_prior = accelerator.prepare(DiffusionPipeline.from_pretrained("warp-ai/wuerstchen-prior", torch_dtype=torch.bfloat16, use_safetensors=True, safety_cheker=None))
|
14 |
warp_prior.scheduler = DDPMScheduler.from_config(warp_prior.scheduler.config)
|
15 |
warp_prior = warp_prior.to("cpu")
|
16 |
+
warp = accelerator.prepare(DiffusionPipeline.from_pretrained("warp-ai/wuerstchen", torch_dtype=torch.bfloat16, use_safetensors=True, safety_checker=None))
|
17 |
warp.scheduler = DDPMScheduler.from_config(warp.scheduler.config)
|
18 |
warp = warp.to("cpu")
|
19 |
generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4876364))
|