Spaces:
Running
on
Zero
Running
on
Zero
Configure for ZERO space
#4
by
Fabrice-TIERCELIN
- opened
app.py
CHANGED
|
@@ -21,6 +21,7 @@ pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|
|
| 21 |
|
| 22 |
max_64_bit_int = 2**63 - 1
|
| 23 |
|
|
|
|
| 24 |
def sample(
|
| 25 |
image: Image,
|
| 26 |
seed: Optional[int] = 42,
|
|
|
|
| 21 |
|
| 22 |
max_64_bit_int = 2**63 - 1
|
| 23 |
|
| 24 |
+
@spaces.GPU(duration=250)
|
| 25 |
def sample(
|
| 26 |
image: Image,
|
| 27 |
seed: Optional[int] = 42,
|