Skip to content

Commit 0bb3f04

Browse files
committed
add more comments
- add TODO for embedding layers
1 parent 09c0c84 commit 0bb3f04

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/diffusers/modeling_flax_pytorch_utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,9 @@ def rename_key_and_reshape_tensor(
5454
if any("norm" in str_ for str_ in pt_tuple_key) and pt_tuple_key[-1] == "weight":
5555
return renamed_pt_tuple_key, pt_tensor
5656

57-
# # embedding
58-
# renamed_pt_tuple_key = pt_tuple_key[:-1] + ("embedding",)
57+
# embedding
58+
# For now the embedding layers are not converted
59+
# TODO: figure out how to detect embedding layers
5960

6061
# conv layer
6162
renamed_pt_tuple_key = pt_tuple_key[:-1] + ("kernel",)

0 commit comments

Comments
 (0)