Spaces:
Running
on
Zero
Running
on
Zero
Aditya Sahu
commited on
Add @spaces.GPU for ZeroGPU
Browse files
app.py
CHANGED
|
@@ -27,6 +27,8 @@ def _resolve_uploaded_path(uploaded):
|
|
| 27 |
return getattr(uploaded, attr)
|
| 28 |
return None
|
| 29 |
|
|
|
|
|
|
|
| 30 |
def compile_model(model_name, vmem_value, lpmem_value, uploaded_model):
|
| 31 |
# Decide the source model path (uploaded has priority)
|
| 32 |
uploaded_path = _resolve_uploaded_path(uploaded_model)
|
|
|
|
| 27 |
return getattr(uploaded, attr)
|
| 28 |
return None
|
| 29 |
|
| 30 |
+
|
| 31 |
+
@spaces.GPU
|
| 32 |
def compile_model(model_name, vmem_value, lpmem_value, uploaded_model):
|
| 33 |
# Decide the source model path (uploaded has priority)
|
| 34 |
uploaded_path = _resolve_uploaded_path(uploaded_model)
|