fix model parallel issues for deimv2 (#45843)
* fix model parallel issues for deimv2
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* refine code
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* update
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* make Deimv2ConvEncoder inherit Deimv2PreTrainedModel
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* Update src/transformers/models/deimv2/modular_deimv2.py
Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>
* add and update commments
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* fix LINT issue
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>
