a little error

[Lisennlp]

Thanks for your contribution, I found a small error in the code：

maxtext/MaxText/train.py

Line 267 in 121293b

metrics_dict = intermediate_outputs["intermediates"]["decoder"]["decoder"]

metrics_dict = intermediate_outputs["intermediates"]["decoder"]["decoder"]
should be:
metrics_dict = intermediate_outputs["intermediates"]["decoder"]["layers"]
？

[shralex]

@Lisennlp thank you for reporting this issue! we were curious whether you use this code ? can you provide some info about your use-case