transformers
1b280f99 - fix device dismatch issue for pe_audio_video model parallelism (#42917)

Commit
110 days ago
fix device dismatch issue for pe_audio_video model parallelism (#42917) * fix device dismatch issue for pe_audio_video model parallelism Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * skip the model parallelism unit test Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * update Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * Set all padding_mask_videos to 1 to avoid NaN values in outputs Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * fix bug Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * keep at least 1 valid frame Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * fix format issue Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * use random.seed 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