Omnibus commited on
Commit
9f5126a
1 Parent(s): 5b2cf12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ repo = "ByteDance/SDXL-Lightning"
13
  ckpt = "sdxl_lightning_4step_unet.safetensors" # Use the correct ckpt for your step setting!
14
  # Load model.
15
  pipe_box=[]
16
- @spaces.GPU
17
  def init():
18
  device="cuda:0"
19
  unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device, torch.float16)
 
13
  ckpt = "sdxl_lightning_4step_unet.safetensors" # Use the correct ckpt for your step setting!
14
  # Load model.
15
  pipe_box=[]
16
+ @spaces.GPU()
17
  def init():
18
  device="cuda:0"
19
  unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device, torch.float16)