Spaces:
Runtime error
Runtime error
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)
|