guowenxiang commited on
Commit
0fac373
1 Parent(s): 14db764

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ class GenSamples:
98
 
99
  return record_dicts
100
 
101
- # @spaces.GPU(enable_queue=True)
102
  def infer(ori_prompt, ddim_steps, scale, seed):
103
  # np.random.seed(seed)
104
  # torch.manual_seed(seed)
 
98
 
99
  return record_dicts
100
 
101
+ @spaces.GPU(enable_queue=True)
102
  def infer(ori_prompt, ddim_steps, scale, seed):
103
  # np.random.seed(seed)
104
  # torch.manual_seed(seed)