Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -122,7 +122,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
| 122 |
seed = random.randint(0, MAX_SEED)
|
| 123 |
return seed
|
| 124 |
|
| 125 |
-
@spaces.GPU(duration=
|
| 126 |
def generate(
|
| 127 |
model_choice: str,
|
| 128 |
prompt: str,
|
|
@@ -132,8 +132,8 @@ def generate(
|
|
| 132 |
seed: int = 1,
|
| 133 |
width: int = 768,
|
| 134 |
height: int = 768,
|
| 135 |
-
guidance_scale: float =
|
| 136 |
-
num_inference_steps: int =
|
| 137 |
randomize_seed: bool = False,
|
| 138 |
use_resolution_binning: bool = True,
|
| 139 |
num_images: int = 1,
|
|
@@ -180,8 +180,8 @@ def generate_cpu(
|
|
| 180 |
seed: int = 1,
|
| 181 |
width: int = 768,
|
| 182 |
height: int = 768,
|
| 183 |
-
guidance_scale: float =
|
| 184 |
-
num_inference_steps: int =
|
| 185 |
randomize_seed: bool = False,
|
| 186 |
use_resolution_binning: bool = True,
|
| 187 |
num_images: int = 1,
|
|
|
|
| 122 |
seed = random.randint(0, MAX_SEED)
|
| 123 |
return seed
|
| 124 |
|
| 125 |
+
@spaces.GPU(duration=80)
|
| 126 |
def generate(
|
| 127 |
model_choice: str,
|
| 128 |
prompt: str,
|
|
|
|
| 132 |
seed: int = 1,
|
| 133 |
width: int = 768,
|
| 134 |
height: int = 768,
|
| 135 |
+
guidance_scale: float = 5,
|
| 136 |
+
num_inference_steps: int = 225,
|
| 137 |
randomize_seed: bool = False,
|
| 138 |
use_resolution_binning: bool = True,
|
| 139 |
num_images: int = 1,
|
|
|
|
| 180 |
seed: int = 1,
|
| 181 |
width: int = 768,
|
| 182 |
height: int = 768,
|
| 183 |
+
guidance_scale: float = 5,
|
| 184 |
+
num_inference_steps: int = 225,
|
| 185 |
randomize_seed: bool = False,
|
| 186 |
use_resolution_binning: bool = True,
|
| 187 |
num_images: int = 1,
|