ford442 commited on
Commit
3a41198
·
1 Parent(s): 5056695

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=75)
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 = 420,
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,