Spaces:
Paused
Paused
Tonic commited on
Commit ·
5cb366f
1
Parent(s): 762264d
removes model loading using ZeroGPU
Browse files
app.py
CHANGED
|
@@ -51,7 +51,7 @@ def clear_gpu_memory():
|
|
| 51 |
torch.cuda.empty_cache()
|
| 52 |
gc.collect()
|
| 53 |
|
| 54 |
-
@spaces.GPU(duration=180)
|
| 55 |
def load_pipeline():
|
| 56 |
"""Load the Chronos model with GPU configuration"""
|
| 57 |
global pipeline
|
|
|
|
| 51 |
torch.cuda.empty_cache()
|
| 52 |
gc.collect()
|
| 53 |
|
| 54 |
+
# @spaces.GPU(duration=180)
|
| 55 |
def load_pipeline():
|
| 56 |
"""Load the Chronos model with GPU configuration"""
|
| 57 |
global pipeline
|