Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -149,7 +149,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
| 149 |
seed = random.randint(0, MAX_SEED)
|
| 150 |
return seed
|
| 151 |
|
| 152 |
-
@spaces.GPU(duration=
|
| 153 |
def generate(
|
| 154 |
model_choice: str,
|
| 155 |
prompt: str,
|
|
@@ -160,7 +160,7 @@ def generate(
|
|
| 160 |
width: int = 768,
|
| 161 |
height: int = 768,
|
| 162 |
guidance_scale: float = 5,
|
| 163 |
-
num_inference_steps: int =
|
| 164 |
randomize_seed: bool = False,
|
| 165 |
use_resolution_binning: bool = True,
|
| 166 |
num_images: int = 1,
|
|
|
|
| 149 |
seed = random.randint(0, MAX_SEED)
|
| 150 |
return seed
|
| 151 |
|
| 152 |
+
@spaces.GPU(duration=60)
|
| 153 |
def generate(
|
| 154 |
model_choice: str,
|
| 155 |
prompt: str,
|
|
|
|
| 160 |
width: int = 768,
|
| 161 |
height: int = 768,
|
| 162 |
guidance_scale: float = 5,
|
| 163 |
+
num_inference_steps: int = 400,
|
| 164 |
randomize_seed: bool = False,
|
| 165 |
use_resolution_binning: bool = True,
|
| 166 |
num_images: int = 1,
|