Browse Source

å修改set_env_on_import的è°执行逻辑

tags/v1.0.0beta
x54-729 2 years ago
parent
commit
b404293743
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      fastNLP/envs/set_env_on_import.py

+ 2
- 1
fastNLP/envs/set_env_on_import.py View File

@@ -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()


Loading…
Cancel
Save