Spaces:
Running
on
Zero
Running
on
Zero
update gpu duration
Browse files
app.py
CHANGED
@@ -194,7 +194,7 @@ def run_midi(
|
|
194 |
return outputs
|
195 |
|
196 |
|
197 |
-
@spaces.GPU(duration=
|
198 |
@torch.no_grad()
|
199 |
@torch.autocast(device_type=DEVICE, dtype=torch.bfloat16)
|
200 |
def run_generation(
|
|
|
194 |
return outputs
|
195 |
|
196 |
|
197 |
+
@spaces.GPU(duration=180)
|
198 |
@torch.no_grad()
|
199 |
@torch.autocast(device_type=DEVICE, dtype=torch.bfloat16)
|
200 |
def run_generation(
|