src/diffusers/models/unets/unet_2d_condition.py · d7a1a0363feb7694568695394a7edfdd44a18bca · renzhc / diffusers_dcu

"src/partition/cuda/partition_op.hip" did not exist on "96297fb8fdd629fc2fd52872d718d78fa491cba1"

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

unet_2d_condition.py 65.5 KB

Replace unet_2d_condition.py