Skip to content

Commit 3d3aae3

Browse files
author
J石页
committed
NPU Adaption for Sanna
1 parent 510e1d6 commit 3d3aae3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/sana_transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,14 +257,14 @@ def __init__(
257257
inner_dim = num_attention_heads * attention_head_dim
258258

259259
# 1. Patch Embedding
260-
interpolation_scale = interpolation_scale if interpolation_scale is not None else max(sample_size // 64, 1)
261260
self.patch_embed = PatchEmbed(
262261
height=sample_size,
263262
width=sample_size,
264263
patch_size=patch_size,
265264
in_channels=in_channels,
266265
embed_dim=inner_dim,
267266
interpolation_scale=interpolation_scale,
267+
pos_embed_type="sincos" if interpolation_scale is not None else None,
268268
)
269269

270270
# 2. Additional condition embeddings

0 commit comments

Comments
 (0)