🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
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>
K
kaixuanliu committed
f403463bd374103a100c9e5f07f717f2aaa37817
Parent: 1ae5aaa
Committed by GitHub <noreply@github.com>
on 5/18/2026, 7:12:44 AM