"tests/models/vscode:/vscode.git/clone" did not exist on "ccdabc5642bf84849af93f591e207dc625c8e1e1"
Unverified Commit 8afaaa26 authored by Ahmed Elnaggar's avatar Ahmed Elnaggar Committed by GitHub
Browse files

FIx Iterations for decoder (#16934)

FIx Iterations for decoder
parent fa322474
......@@ -100,7 +100,7 @@ def convert_t5x_checkpoint_to_flax(t5x_checkpoint_path, config_name, flax_dump_f
flax_model.params["encoder"]["final_layer_norm"]["weight"] = t5x_encoder_norm
# Decoder
for layer_index in range(config.num_layers):
for layer_index in range(config.num_decoder_layers):
layer_name = f"layers_{str(layer_index)}"
# Self-Attention
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment