Kosmos-2 Fine tuning

🤗Transformers
26

@Mit1208 I just changed add a line before 1152 at modeling_kosmos2.py to temporarily enable fp16

inputs_embeds = inputs_embeds.to(image_embeds.dtype)

2 Likes