Update app.py
Browse files
app.py
CHANGED
|
@@ -21,9 +21,9 @@ pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo_id,
|
|
| 21 |
pipe = pipe.to(device)
|
| 22 |
|
| 23 |
MAX_SEED = np.iinfo(np.int32).max
|
| 24 |
-
MAX_IMAGE_SIZE =
|
| 25 |
|
| 26 |
-
@spaces.GPU(duration=
|
| 27 |
def infer(
|
| 28 |
prompt,
|
| 29 |
negative_prompt="",
|
|
|
|
| 21 |
pipe = pipe.to(device)
|
| 22 |
|
| 23 |
MAX_SEED = np.iinfo(np.int32).max
|
| 24 |
+
MAX_IMAGE_SIZE = 2048
|
| 25 |
|
| 26 |
+
@spaces.GPU(duration=25)
|
| 27 |
def infer(
|
| 28 |
prompt,
|
| 29 |
negative_prompt="",
|