Skip to content

Commit dc26828

Browse files
To avoid recursive import temporarily (#1896)
* To avoid recursive import temporarily. * To avoid recursive import temporarily. Co-authored-by: Jiaqi Liu <[email protected]>
1 parent 86489d8 commit dc26828

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddlenlp/transformers/model_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@
2727
from paddlenlp.utils.downloader import get_path_from_url, COMMUNITY_MODEL_PREFIX
2828
from paddlenlp.utils.env import MODEL_HOME
2929
from paddlenlp.utils.log import logger
30-
import paddlenlp.ops.faster_transformer.transformer.decoding as ft_decoding
3130

3231
from .generation_utils import GenerationMixin
3332
from .utils import InitTrackerMeta, fn_args_to_dict
@@ -389,6 +388,8 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
389388
if k in state_to_load:
390389
state_to_load[k] = state_to_load[k].astype(dtype)
391390
# For model parallel if FasterGeneration
391+
# To avoid recursive import temporarily.
392+
import paddlenlp.ops.faster_transformer.transformer.decoding as ft_decoding
392393
state_to_load = ft_decoding.get_ft_para_conf().fit_partial_model(
393394
model_to_load, state_to_load)
394395
if paddle.in_dynamic_mode():

0 commit comments

Comments
 (0)