Muinez commited on
Commit
ba84c88
·
verified ·
1 Parent(s): 7e54fb9

Add duration param to save quota

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ class Pipeline:
46
  def __del__(self):
47
  self.tmp_dir.cleanup()
48
 
49
- @spaces.GPU
50
  @torch.no_grad()
51
  @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
52
  def __call__(self, y, seed, num_steps, guidance, timeshift, order):
 
46
  def __del__(self):
47
  self.tmp_dir.cleanup()
48
 
49
+ @spaces.GPU(duration=5)
50
  @torch.no_grad()
51
  @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
52
  def __call__(self, y, seed, num_steps, guidance, timeshift, order):