gokaygokay
/

Florence-2-Flux-Large

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

gokaygokay commited on Aug 25

Commit

d114049

•

1 Parent(s): db345b8

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -35,7 +35,8 @@ def run_example(task_prompt, text_input, image):
  input_ids=inputs["input_ids"],
  pixel_values=inputs["pixel_values"],
  max_new_tokens=1024,
- num_beams=3
  )
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
  parsed_answer = processor.post_process_generation(generated_text, task=task_prompt, image_size=(image.width, image.height))

  input_ids=inputs["input_ids"],
  pixel_values=inputs["pixel_values"],
  max_new_tokens=1024,
+ num_beams=3,
+ repetition_penalty=1.10,
  )
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
  parsed_answer = processor.post_process_generation(generated_text, task=task_prompt, image_size=(image.width, image.height))