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>