From e87f39e289a5461c2c0778664bf51fb175aba02f Mon Sep 17 00:00:00 2001 From: Joseph Attieh Date: Tue, 6 Feb 2024 11:54:50 +0200 Subject: [PATCH] Fixed changes for PR --- mammoth/model_builder.py | 3 ++- mammoth/opts.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/mammoth/model_builder.py b/mammoth/model_builder.py index 578c8cb3..71f6d1ce 100644 --- a/mammoth/model_builder.py +++ b/mammoth/model_builder.py @@ -49,7 +49,8 @@ def build_embeddings(opts, vocab, for_encoder=True): freeze_word_vecs=freeze_word_vecs, enable_embeddingless=opts.enable_embeddingless ) - if opts.enable_embeddingless: logger.info("Creating an embeddingless model.") + if opts.enable_embeddingless: + logger.info("Creating an embeddingless model.") return emb diff --git a/mammoth/opts.py b/mammoth/opts.py index a18eb4e7..f536a8ea 100644 --- a/mammoth/opts.py +++ b/mammoth/opts.py @@ -330,8 +330,8 @@ def model_opts(parser): '--enable_embeddingless', '-enable_embeddingless', action='store_true', - help="Enable the use of byte-based embeddingless models" + \ - "(Shaham et. al, 2021) https://aclanthology.org/2021.naacl-main.17/", + help="Enable the use of byte-based embeddingless models" + + "(Shaham et. al, 2021) https://aclanthology.org/2021.naacl-main.17/", ) group.add( '--position_encoding',