[model][NPU]:Z-image model support NPU

2026-03-18 22:08:13 +00:00 · 2026-01-07 11:42:19 +08:00
parent 3ee5f53a36
commit c1c9a4853b
1 changed files with 2 additions and 2 deletions
--- a/diffsynth/models/z_image_dit.py
+++ b/diffsynth/models/z_image_dit.py
@@ -276,9 +276,9 @@ class RopeEmbedder:
        for i in range(len(self.axes_dims)):
            index = ids[:, i]
            if IS_NPU_AVAILABLE:
-                result.append(self.freqs_cis[i][index])
-            else:
                result.append(torch.index_select(self.freqs_cis[i], 0, index))
+            else:
+                result.append(self.freqs_cis[i][index])
        return torch.cat(result, dim=-1)