Fix AuraFlow model parallelism device mismatch and update XPU IP-Adap… (#14273)
* Fix AuraFlow model parallelism device mismatch and update XPU IP-Adapter expectations
Description
This PR fixes two test issues currently reflected in the working tree.
Changes
Fixed AuraFlow model parallelism in auraflow_transformer_2d.py.
Under device_map="auto", context_embedder can place encoder_hidden_states on a different device from the top-level register_tokens parameter. The forward pass now moves register_tokens to the projected encoder hidden states device and dtype before concatenation, avoiding cross-device torch.cat failures.
Updated XPU-specific expected slices in test_ip_adapter_stable_diffusion.py.
The IP-Adapter SD integration expectation now distinguishes XPU backend versions 3 and 5, since they produce slightly different deterministic slices.
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
* update
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
* update
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
* fmt
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
---------
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>