Hugging Face's logo

Spaces:
briaai
/
BRIA-2.2-FAST
Running on Zero

App Files Community
1
Eyalgut commited on
Commit
30d0900
·
verified ·
1 Parent(s): d9daae5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ class Dummy():
11
 
12
  resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280" ]
13
 
14
- # Load pipeline
15
 
16
  unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-LCM-2.2", torch_dtype=torch.float16)
17
  pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.2", unet=unet, torch_dtype=torch.float16)
 
11
 
12
  resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280" ]
13
 
14
+ # Load pipeline
15
 
16
  unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-LCM-2.2", torch_dtype=torch.float16)
17
  pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.2", unet=unet, torch_dtype=torch.float16)