diff --git a/fastNLP/envs/imports.py b/fastNLP/envs/imports.py index c0ffffa3..08afc6a5 100644 --- a/fastNLP/envs/imports.py +++ b/fastNLP/envs/imports.py @@ -22,10 +22,7 @@ _NEED_IMPORT_FAIRSCALE = not _IS_WINDOWS and _module_available("fairscale") and _NEED_IMPORT_TORCH = _module_available("torch") and 'torch' in need_import _NEED_IMPORT_JITTOR = _module_available("jittor") and 'jittor' in need_import _NEED_IMPORT_PADDLE = _module_available("paddle") and 'paddle' in need_import -<<<<<<< HEAD _NEED_IMPORT_DEEPSPEED = _module_available("deepspeed") and 'torch' in need_import -======= _NEED_IMPORT_ONEFLOW = _module_available("oneflow") and 'oneflow' in need_import ->>>>>>> dev0.8.0 _TORCH_GREATER_EQUAL_1_8 = _NEED_IMPORT_TORCH and _compare_version("torch", operator.ge, "1.8.0")