modify some style

This commit is contained in:
BUAADreamer committed 2024-04-25 21:58:18 +08:00
1 parent 2cab2d42fb
commit 2d4ded535f
6 files changed
+26 -158

No files matched your search

+1 -3
View File
@@ -41,9 +41,7 @@ def _get_init_kwargs(model_args: "ModelArguments") -> Dict[str, Any]:
}
def load_tokenizer(
model_args: "ModelArguments",
) -> Dict[str, Union["PreTrainedTokenizer", "AutoProcessor"]]:
def load_tokenizer(model_args: "ModelArguments") -> Dict[str, Union["PreTrainedTokenizer", "AutoProcessor"]]:
r"""
Loads pretrained tokenizer.