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,
|