Spaces:
Running
Running
Ming Li
commited on
Commit
•
e8acbbf
1
Parent(s):
68f63f3
rm zero gpu on run_pipe func
Browse files
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,
|