jayparmr commited on
Commit
fe95d1e
1 Parent(s): bcaef47

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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())] * num_return_sequences
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)