Spaces:
Paused
Paused
Tonic commited on
Commit ·
a96f134
1
Parent(s): 0695ae5
add decorator to load_pipeline for ZerGPU
Browse files
app.py
CHANGED
|
@@ -51,6 +51,7 @@ def clear_gpu_memory():
|
|
| 51 |
torch.cuda.empty_cache()
|
| 52 |
gc.collect()
|
| 53 |
|
|
|
|
| 54 |
def load_pipeline():
|
| 55 |
"""Load the Chronos model without GPU configuration"""
|
| 56 |
global pipeline
|
|
|
|
| 51 |
torch.cuda.empty_cache()
|
| 52 |
gc.collect()
|
| 53 |
|
| 54 |
+
@spaces.GPU()
|
| 55 |
def load_pipeline():
|
| 56 |
"""Load the Chronos model without GPU configuration"""
|
| 57 |
global pipeline
|