Skip to content

Commit

Permalink
Removing unnecessary +1 in the module name
Browse files Browse the repository at this point in the history
  • Loading branch information
NohTow committed Aug 12, 2024
1 parent 2e72d68 commit a0dfc74
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion giga_cherche/models/colbert.py
Original file line number Diff line number Diff line change
Expand Up @@ -251,7 +251,7 @@ def __init__(
f"The checkpoint does not contain a linear projection layer. Adding one with output dimensions ({hidden_size}, {embedding_size})"
)
# Add a linear projection layer to the model in order to project the embeddings to the desired size
self._modules[f"{len(self._modules) + 1}"] = LinearProjection(
self._modules[f"{len(self._modules)}"] = LinearProjection(
in_features=hidden_size, out_features=embedding_size, bias=False
)

Expand Down

0 comments on commit a0dfc74

Please sign in to comment.