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