Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ resolutions = ["1536 1536","1728 1280","1856 1280","1920-1088", "1088 1920","128
|
|
16 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
17 |
unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-2.2-HR", torch_dtype=torch.float16)
|
18 |
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.2", torch_dtype=torch.float16, unet=unet, vae=vae)
|
19 |
-
|
20 |
pipe.to('cuda')
|
21 |
del unet
|
22 |
del vae
|
|
|
16 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
17 |
unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-2.2-HR", torch_dtype=torch.float16)
|
18 |
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.2", torch_dtype=torch.float16, unet=unet, vae=vae)
|
19 |
+
assert type(pipe.scheduler) == EulerAncestralDiscreteScheduler
|
20 |
pipe.to('cuda')
|
21 |
del unet
|
22 |
del vae
|