transformers
760946c7 - handle 1D position_ids for modeling_flash_attention_utils as well (#43403)

Commit
184 days ago
handle 1D position_ids for modeling_flash_attention_utils as well (#43403) * handle 1D position_ids for modeling_flash_attention_utils as well Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * delete comments Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * fixed unit test of lighton_ocr Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * add image_sizes to first_inputs Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * avoid change base class Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * update code Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * update code Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> --------- Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
Author
Parents
Loading