Spaces:
Running
on
Zero
Running
on
Zero
Avijit Ghosh
commited on
Commit
·
b9bfe79
1
Parent(s):
de81f33
add gpu wrapper
Browse files
app.py
CHANGED
|
@@ -13,7 +13,6 @@ import os
|
|
| 13 |
import spaces
|
| 14 |
|
| 15 |
# Define model initialization functions
|
| 16 |
-
@spaces.GPU
|
| 17 |
def load_model(model_name):
|
| 18 |
if model_name == "stabilityai/sdxl-turbo":
|
| 19 |
pipeline = DiffusionPipeline.from_pretrained(
|
|
|
|
| 13 |
import spaces
|
| 14 |
|
| 15 |
# Define model initialization functions
|
|
|
|
| 16 |
def load_model(model_name):
|
| 17 |
if model_name == "stabilityai/sdxl-turbo":
|
| 18 |
pipeline = DiffusionPipeline.from_pretrained(
|