Refactor: unify invoke format of open() function

This commit is contained in:
tsukumi
2024-03-12 18:00:51 +00:00
parent 74af2c831c
commit 483bc68d57
9 changed files with 24 additions and 21 deletions

View File

@@ -39,7 +39,7 @@ def load_model(
language: Languages,
pretrained_model_name_or_path: Optional[str] = None,
cache_dir: Optional[str] = None,
revision: str = 'main',
revision: str = "main",
) -> Union[PreTrainedModel, DebertaV2Model]:
"""
指定された言語の BERT モデルをロードし、ロード済みの BERT モデルを返す。
@@ -96,7 +96,7 @@ def load_tokenizer(
language: Languages,
pretrained_model_name_or_path: Optional[str] = None,
cache_dir: Optional[str] = None,
revision: str = 'main',
revision: str = "main",
) -> Union[PreTrainedTokenizer, PreTrainedTokenizerFast, DebertaV2Tokenizer]:
"""
指定された言語の BERT モデルをロードし、ロード済みの BERT トークナイザーを返す。