magnetism4236 commited on
Commit
73984fa
·
verified ·
1 Parent(s): 3f34c8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -246,9 +246,8 @@ def infer(
246
  prompt,
247
  lora_choice,
248
  seed=42,
249
- randomize_seed=False,
250
  true_guidance_scale=4.0,
251
- num_inference_steps=40,
252
  height=None,
253
  width=None,
254
  # rewrite_prompt=False,
@@ -292,6 +291,9 @@ def infer(
292
  if pil_images:
293
  for i, img in enumerate(pil_images):
294
  print(f" [{i}] size: {img.width}x{img.height}")
 
 
 
295
  images = pipe(
296
  image=pil_images if len(pil_images) > 0 else None,
297
  prompt=prompt,
 
246
  prompt,
247
  lora_choice,
248
  seed=42,
 
249
  true_guidance_scale=4.0,
250
+ num_inference_steps=8,
251
  height=None,
252
  width=None,
253
  # rewrite_prompt=False,
 
291
  if pil_images:
292
  for i, img in enumerate(pil_images):
293
  print(f" [{i}] size: {img.width}x{img.height}")
294
+
295
+ num_inference_steps = max(1, int(num_inference_steps or 1))
296
+
297
  images = pipe(
298
  image=pil_images if len(pil_images) > 0 else None,
299
  prompt=prompt,