Spaces:
Runtime error
Runtime error
Commit
•
8fcb7b8
1
Parent(s):
9b3746f
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
|
| 40 |
#pipe.enable_model_cpu_offload()
|
| 41 |
pipe.to('cuda')
|
| 42 |
|
| 43 |
-
@spaces.GPU
|
| 44 |
def core(bundle):
|
| 45 |
generator = torch.Generator().manual_seed(int(bundle["seed"]))
|
| 46 |
result = pipe(
|
|
|
|
| 40 |
#pipe.enable_model_cpu_offload()
|
| 41 |
pipe.to('cuda')
|
| 42 |
|
| 43 |
+
@spaces.GPU(duration=90)
|
| 44 |
def core(bundle):
|
| 45 |
generator = torch.Generator().manual_seed(int(bundle["seed"]))
|
| 46 |
result = pipe(
|