diff --git a/deepmd/common.py b/deepmd/common.py
index 3ab936db67..5581f3533b 100644
--- a/deepmd/common.py
+++ b/deepmd/common.py
@@ -44,7 +44,7 @@
     "select_idx_map",
 ]
 
-_PRECISION = Literal["default", "float16", "float32", "float64"]
+_PRECISION = Literal["default", "float16", "bfloat16", "float32", "float64"]
 _ACTIVATION = Literal[
     "relu",
     "relu6",