From b40429374399f49371fc76ab455b8543acf644d5 Mon Sep 17 00:00:00 2001 From: x54-729 <17307130121@fudan.edu.cn> Date: Sun, 10 Jul 2022 12:09:05 +0000 Subject: [PATCH] =?UTF-8?q?=C3=A5=E4=BF=AE=E6=94=B9set=5Fenv=5Fon=5Fimport?= =?UTF-8?q?=E7=9A=84=C3=A8=C2=B0=E6=89=A7=E8=A1=8C=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/envs/set_env_on_import.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/fastNLP/envs/set_env_on_import.py b/fastNLP/envs/set_env_on_import.py index c77e7402..b4258154 100644 --- a/fastNLP/envs/set_env_on_import.py +++ b/fastNLP/envs/set_env_on_import.py @@ -68,7 +68,8 @@ def set_env_on_import(): :return: """ # 框架相关的变量设置 - set_env_on_import_torch() + if "oneflow" not in sys.modules: + set_env_on_import_torch() set_env_on_import_paddle() set_env_on_import_jittor() set_env_on_import_oneflow()