Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,8 @@ pipe.to(device, torch_dtype)
|
|
| 50 |
MAX_SEED = np.iinfo(np.int32).max
|
| 51 |
MAX_IMAGE_SIZE = 1536
|
| 52 |
|
| 53 |
-
|
|
|
|
| 54 |
def infer(
|
| 55 |
prompt,
|
| 56 |
negative_prompt="",
|
|
|
|
| 50 |
MAX_SEED = np.iinfo(np.int32).max
|
| 51 |
MAX_IMAGE_SIZE = 1536
|
| 52 |
|
| 53 |
+
# Increase maximum duration for people who want to make large images
|
| 54 |
+
@spaces.GPU(duration=120)
|
| 55 |
def infer(
|
| 56 |
prompt,
|
| 57 |
negative_prompt="",
|