diff --git a/deepmd/pt/utils/env.py b/deepmd/pt/utils/env.py index 5616f0a9a5..0b92953255 100644 --- a/deepmd/pt/utils/env.py +++ b/deepmd/pt/utils/env.py @@ -68,6 +68,7 @@ "GLOBAL_ENER_FLOAT_PRECISION", "GLOBAL_NP_FLOAT_PRECISION", "GLOBAL_PT_FLOAT_PRECISION", + "GLOBAL_PT_ENER_FLOAT_PRECISION", "DEFAULT_PRECISION", "PRECISION_DICT", "RESERVED_PRECISON_DICT",