Spaces:
Runtime error
Runtime error
Ming Li
commited on
Commit
·
43208f4
1
Parent(s):
f07cca1
update requirements.txt
Browse files
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,
|