Skip to content

Commit c1c9a48

Browse files
committed
[model][NPU]:Z-image model support NPU
1 parent 3ee5f53 commit c1c9a48

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

diffsynth/models/z_image_dit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,9 +276,9 @@ def __call__(self, ids: torch.Tensor):
276276
for i in range(len(self.axes_dims)):
277277
index = ids[:, i]
278278
if IS_NPU_AVAILABLE:
279-
result.append(self.freqs_cis[i][index])
280-
else:
281279
result.append(torch.index_select(self.freqs_cis[i], 0, index))
280+
else:
281+
result.append(self.freqs_cis[i][index])
282282
return torch.cat(result, dim=-1)
283283

284284

0 commit comments

Comments
 (0)