Skip to content

Commit

Permalink
Modifying logging to be more accurate
Browse files Browse the repository at this point in the history
  • Loading branch information
NohTow committed Sep 12, 2024
1 parent 54788cf commit bd02ff9
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion pylate/models/colbert.py
Original file line number Diff line number Diff line change
Expand Up @@ -272,6 +272,7 @@ def __init__(
logger.warning(
f"The checkpoint does not contain a linear projection layer. Adding one with output dimensions ({hidden_size}, {embedding_size})."
)
logger.warning("Created a PyLate model from base encoder.")
self.append(
Dense(
in_features=hidden_size, out_features=embedding_size, bias=bias
Expand Down Expand Up @@ -1079,7 +1080,7 @@ def _load_auto_model(
"""
logger.warning(
f"No sentence-transformers model found with name {model_name_or_path}. Creating a ColBERT model from base encoder."
f"No sentence-transformers model found with name {model_name_or_path}."
)

shared_kwargs = {
Expand Down

0 comments on commit bd02ff9

Please sign in to comment.