Skip to content

Commit 6de32e5

Browse files
authored
rename download_check command (#2134)
1 parent 103241b commit 6de32e5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlenlp/transformers/model_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -388,7 +388,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
388388
if k in state_to_load:
389389
state_to_load[k] = state_to_load[k].astype(dtype)
390390
# Logging model download statistics
391-
download_check(pretrained_model_name_or_path, cls.__name__)
391+
download_check(pretrained_model_name_or_path, "from_pretrained")
392392
# For model parallel if FasterGeneration
393393
# To avoid recursive import temporarily.
394394
import paddlenlp.ops.faster_transformer.transformer.decoding as ft_decoding

0 commit comments

Comments
 (0)