Spaces:
Running
on
Zero
Running
on
Zero
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
|