Skip to content

Commit

Permalink
black
Browse files Browse the repository at this point in the history
  • Loading branch information
vince62s committed Dec 18, 2024
1 parent 582f7eb commit 85e8420
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion python/ctranslate2/converters/eole_ct2.py
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,9 @@ def set_transformer_encoder(spec, variables):
set_input_layers(spec, variables, "src_emb")
set_layer_norm(spec.layer_norm, variables, "encoder.layer_norm")
for i, layer in enumerate(spec.layer):
set_transformer_encoder_layer(layer, variables, "encoder.transformer_layers.%d" % i)
set_transformer_encoder_layer(
layer, variables, "encoder.transformer_layers.%d" % i
)


def set_transformer_decoder(spec, variables, with_encoder_attention=True):
Expand Down

0 comments on commit 85e8420

Please sign in to comment.