Spaces:
Paused
Paused
Commit
·
5ecdf4c
1
Parent(s):
aba49e2
fix bug
Browse files
app.py
CHANGED
@@ -397,13 +397,13 @@ def prepare_infer_func():
|
|
397 |
|
398 |
return image_edit.generate_image
|
399 |
|
400 |
-
def inference(
|
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,
|