Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280" ]
|
|
15 |
|
16 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
17 |
unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-2.3-FAST", torch_dtype=torch.float16)
|
18 |
-
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3
|
19 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
20 |
pipe.to('cuda')
|
21 |
del unet
|
|
|
15 |
|
16 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
17 |
unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-2.3-FAST", torch_dtype=torch.float16)
|
18 |
+
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3", torch_dtype=torch.float16, unet=unet, vae=vae)
|
19 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
20 |
pipe.to('cuda')
|
21 |
del unet
|