Skip to content

Commit 1a6fca8

Browse files
author
gongel
committed
update log
1 parent 61e8c9b commit 1a6fca8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddlenlp/transformers/bart/converter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@
9494
paddle_state_dict = OrderedDict()
9595

9696
# Convert parameters
97-
for k, v in hf_model.items() if use_torch else hf_model.items():
97+
for k, v in hf_model.items():
9898
transpose = False
9999
if k in skip_weights:
100100
continue
@@ -118,7 +118,7 @@
118118
if oldk == "lm_head.weight":
119119
k = "lm_head_weight"
120120

121-
print(f"Converting: {oldk} => {k} | is_transpose {transpose}")
121+
logger.info(f"Converting: {oldk} => {k} | is_transpose {transpose}")
122122

123123
paddle_state_dict[k] = v.data.numpy() if use_torch else v
124124

0 commit comments

Comments
 (0)