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>