Spaces:
Running
on
Zero
Running
on
Zero
Update injection_main_HF.py
Browse files- injection_main_HF.py +1 -1
injection_main_HF.py
CHANGED
|
@@ -522,7 +522,7 @@ if __name__ == "__main__":
|
|
| 522 |
# Set up a DDIM scheduler
|
| 523 |
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
| 524 |
|
| 525 |
-
@spaces.GPU()
|
| 526 |
def style_transfer_app(
|
| 527 |
prompt,
|
| 528 |
image,
|
|
|
|
| 522 |
# Set up a DDIM scheduler
|
| 523 |
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
| 524 |
|
| 525 |
+
@spaces.GPU(duration=90)
|
| 526 |
def style_transfer_app(
|
| 527 |
prompt,
|
| 528 |
image,
|