Spaces:
Running
Running
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,
|