tests/models/test_modeling_common.py · 14725164be74031905029cc3665956bef151bb5d · renzhc / diffusers_dcu

fix "Expected all tensors to be on the same device, but found at least two devices" error (#11690) · 14725164

Yao Matrix authored Jul 22, 2025



* 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>

14725164

test_modeling_common.py 101 KB

Replace test_modeling_common.py