listen2you003 commited on
Commit
5ecdf4c
·
1 Parent(s): aba49e2
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -397,13 +397,13 @@ def prepare_infer_func():
397
 
398
  return image_edit.generate_image
399
 
400
- def inference(infer_func, prompt, ref_images, seed, size_level):
401
  start_time = time.time()
402
 
403
  image = infer_func(
404
  prompt,
405
  negative_prompt="",
406
- ref_images=ref_images,
407
  num_samples=1,
408
  num_steps=28,
409
  cfg_guidance=6.0,
 
397
 
398
  return image_edit.generate_image
399
 
400
+ def inference(prompt, ref_images, seed, size_level, infer_func=None):
401
  start_time = time.time()
402
 
403
  image = infer_func(
404
  prompt,
405
  negative_prompt="",
406
+ ref_images=ref_images.convert('RGB'),
407
  num_samples=1,
408
  num_steps=28,
409
  cfg_guidance=6.0,