Spaces:
Running
Running
Update app.py
Browse files
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)
|