Tonic commited on
Commit
a96f134
·
1 Parent(s): 0695ae5

add decorator to load_pipeline for ZerGPU

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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