Skip to content

Commit f5f47ad

Browse files
authored
[Trainer] Fix try import (#6691)
1 parent 9040378 commit f5f47ad

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

paddlenlp/trainer/trainer.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,22 @@
4343
)
4444
from paddle.distributed.fleet.meta_optimizers.dygraph_optimizer.hybrid_parallel_optimizer import (
4545
HybridParallelOptimizer,
46-
_obtain_optimizer_parameters_list,
4746
)
47+
48+
try:
49+
from paddle.distributed.fleet.utils.hybrid_parallel_util import (
50+
obtain_optimizer_parameters_list,
51+
)
52+
53+
_obtain_optimizer_parameters_list = obtain_optimizer_parameters_list
54+
except:
55+
try:
56+
from paddle.distributed.fleet.meta_optimizers.dygraph_optimizer.hybrid_parallel_optimizer import (
57+
_obtain_optimizer_parameters_list,
58+
)
59+
except:
60+
_obtain_optimizer_parameters_list = None
61+
4862
from paddle.distributed.fleet.utils.hybrid_parallel_util import (
4963
fused_allreduce_gradients,
5064
)

0 commit comments

Comments
 (0)