Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,7 +70,7 @@ pipeline = PixelFlowPipeline(
|
|
| 70 |
max_token_length=512,
|
| 71 |
)
|
| 72 |
|
| 73 |
-
@spaces.GPU
|
| 74 |
def infer(noise_shift, cfg_scale, class_label, seed, *num_steps_per_stage):
|
| 75 |
print(f"inside space.GPU. {torch.cuda.is_available()=}")
|
| 76 |
seed_everything(seed)
|
|
|
|
| 70 |
max_token_length=512,
|
| 71 |
)
|
| 72 |
|
| 73 |
+
@spaces.GPU(duration=120)
|
| 74 |
def infer(noise_shift, cfg_scale, class_label, seed, *num_steps_per_stage):
|
| 75 |
print(f"inside space.GPU. {torch.cuda.is_available()=}")
|
| 76 |
seed_everything(seed)
|