fix "Expected all tensors to be on the same device, but found at least two devices" error (#11690)
* xx
* fix
Signed-off-by: YAO Matrix <matrix.yao@intel.com>
* Update model_loading_utils.py
* Update test_models_unet_2d_condition.py
* Update test_models_unet_2d_condition.py
* fix style
Signed-off-by: YAO Matrix <matrix.yao@intel.com>
* fix comments
Signed-off-by: Matrix Yao <matrix.yao@intel.com>
* Update unet_2d_blocks.py
* update
Signed-off-by: Matrix Yao <matrix.yao@intel.com>
---------
Signed-off-by: YAO Matrix <matrix.yao@intel.com>
Signed-off-by: Matrix Yao <matrix.yao@intel.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>