fix error: 'BlockMask' object has no attribute 'dtype' for lasr model (#42802)
* fix error: 'BlockMask' object has no attribute 'dtype' for lasr model
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* fix error: 'BlockMask' object has no attribute 'dtype' for lasr model
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* update code
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* disable flex_attn
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* add comment
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* update comment
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
---------
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>