Spaces:
Runtime error
Runtime error
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,
|