Skip to content

Commit 79a1d31

Browse files
committed
fix
1 parent 3ab805a commit 79a1d31

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/convert_sd3_to_diffusers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def get_add_attn2_layers(state_dict):
251251
# Extract the layer number from the key
252252
layer_num = int(key.split(".")[1])
253253
add_attn2_layers.append(layer_num)
254-
return tuple(sorted(add_attn2_layers))
254+
return tuple(sorted(set(add_attn2_layers)))
255255

256256

257257
def get_pos_embed_max_size(state_dict):

0 commit comments

Comments
 (0)