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,
|