Ming Li commited on
Commit
e8acbbf
1 Parent(s): 68f63f3

rm zero gpu on run_pipe func

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