fix device mismatch issue for HiDreamTransformerTests (#13766)
* fix device mismatch issue for HiDreamTransformerTests
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
* refine code
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
---------
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>