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