ford442 commited on
Commit
88e2385
·
1 Parent(s): 9f0ce3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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=75)
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 = 4,
136
- num_inference_steps: int = 150,
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 = 4,
184
- num_inference_steps: int = 150,
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,