theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 01:48:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
feng0w0
2026-01-07 11:42:19 +08:00
parent 3ee5f53a36
commit c1c9a4853b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/models/z_image_dit.py
View File

@@ -276,9 +276,9 @@ class RopeEmbedder:
for i in range(len(self.axes_dims)): for i in range(len(self.axes_dims)):
index = ids[:, i] index = ids[:, i]
if IS_NPU_AVAILABLE: if IS_NPU_AVAILABLE:
result.append(self.freqs_cis[i][index])
else:
result.append(torch.index_select(self.freqs_cis[i], 0, index)) 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) return torch.cat(result, dim=-1)