Eyalgut commited on
Commit
7d8193f
·
verified ·
1 Parent(s): dceaee5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
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