Ming Li commited on
Commit
43208f4
1 Parent(s): f07cca1

update requirements.txt

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -102,7 +102,7 @@ class Model:
102
  return prompt
103
 
104
  @torch.autocast("cuda")
105
- @spaces.GPU
106
  def run_pipe(
107
  self,
108
  prompt: str,
 
102
  return prompt
103
 
104
  @torch.autocast("cuda")
105
+ @spaces.GPU()
106
  def run_pipe(
107
  self,
108
  prompt: str,