This commit is contained in:
ayush chaurasia
2024-04-15 17:11:59 +05:30
parent d0c1113417
commit fd8de238bb
2 changed files with 2 additions and 2 deletions

View File

@@ -214,7 +214,7 @@ class SentenceTransformersTuner(BaseEmbeddingTuner):
def helper(self) -> None:
"""A helper method."""
logging.info("Finetuning complete.")
logging.info(f"Model saved to {self.path}.")
logging.info(f"Model saved to {self.path}.") # noqa
logging.info("You can now use the model as follows:")
logging.info(
f"model = get_registry().get('sentence-transformers').create(name='./{self.path}')" # noqa

View File

@@ -42,4 +42,4 @@ def test_finetuning_sentence_transformers(tmp_path):
def test_text_chunk():
# TODO
pass
pass