Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
•
4d72328
1
Parent(s):
3b23350
add gpu time
Browse files- app_haircolor.py +1 -1
app_haircolor.py
CHANGED
|
@@ -61,7 +61,7 @@ basepipeline = basepipeline.to(DEVICE)
|
|
| 61 |
|
| 62 |
basepipeline.enable_model_cpu_offload()
|
| 63 |
|
| 64 |
-
@spaces.GPU(duration=
|
| 65 |
def image_to_image(
|
| 66 |
input_image: Image,
|
| 67 |
edit_prompt: str,
|
|
|
|
| 61 |
|
| 62 |
basepipeline.enable_model_cpu_offload()
|
| 63 |
|
| 64 |
+
@spaces.GPU(duration=30)
|
| 65 |
def image_to_image(
|
| 66 |
input_image: Image,
|
| 67 |
edit_prompt: str,
|