We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 09c0c84 commit 0bb3f04Copy full SHA for 0bb3f04
src/diffusers/modeling_flax_pytorch_utils.py
@@ -54,8 +54,9 @@ def rename_key_and_reshape_tensor(
54
if any("norm" in str_ for str_ in pt_tuple_key) and pt_tuple_key[-1] == "weight":
55
return renamed_pt_tuple_key, pt_tensor
56
57
- # # embedding
58
- # renamed_pt_tuple_key = pt_tuple_key[:-1] + ("embedding",)
+ # embedding
+ # For now the embedding layers are not converted
59
+ # TODO: figure out how to detect embedding layers
60
61
# conv layer
62
renamed_pt_tuple_key = pt_tuple_key[:-1] + ("kernel",)
0 commit comments