Spaces:
Running
on
Zero
Running
on
Zero
AbstractPhil
commited on
Commit
·
06ab384
1
Parent(s):
0631943
300s duration removed
Browse files
app.py
CHANGED
|
@@ -155,7 +155,7 @@ class RoseGuidedLogits(torch.nn.Module):
|
|
| 155 |
# -----------------------
|
| 156 |
# ZeroGPU entrypoint (must exist at import time)
|
| 157 |
# -----------------------
|
| 158 |
-
@spaces.GPU(
|
| 159 |
def zerogpu_generate(full_prompt: str,
|
| 160 |
gen_kwargs: Dict[str, Any],
|
| 161 |
rose_map: Optional[Dict[str, float]],
|
|
|
|
| 155 |
# -----------------------
|
| 156 |
# ZeroGPU entrypoint (must exist at import time)
|
| 157 |
# -----------------------
|
| 158 |
+
@spaces.GPU()
|
| 159 |
def zerogpu_generate(full_prompt: str,
|
| 160 |
gen_kwargs: Dict[str, Any],
|
| 161 |
rose_map: Optional[Dict[str, float]],
|