Upload folder using huggingface_hub
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -550,7 +550,7 @@ def img2img(task: Task):
|
|
| 550 |
def inpaint(task: Task):
|
| 551 |
if task.get_type() == TaskType.OUTPAINT:
|
| 552 |
key = "_outpaint"
|
| 553 |
-
prompt = [img2text.process(task.get_imageUrl())] *
|
| 554 |
else:
|
| 555 |
key = "_inpaint"
|
| 556 |
prompt, _ = get_patched_prompt(task)
|
|
|
|
| 550 |
def inpaint(task: Task):
|
| 551 |
if task.get_type() == TaskType.OUTPAINT:
|
| 552 |
key = "_outpaint"
|
| 553 |
+
prompt = [img2text.process(task.get_imageUrl())] * get_num_return_sequences()
|
| 554 |
else:
|
| 555 |
key = "_inpaint"
|
| 556 |
prompt, _ = get_patched_prompt(task)
|