diff --git a/mindpet/delta/delta_constants.py b/mindpet/delta/delta_constants.py index e05a709..c271f06 100755 --- a/mindpet/delta/delta_constants.py +++ b/mindpet/delta/delta_constants.py @@ -5,4 +5,4 @@ from mindspore import dtype as mstype # 算法精度与类型控制 -VALID_TENSOR_DATATYPE = [mstype.float16, mstype.float32] +VALID_TENSOR_DATATYPE = [mstype.float16, mstype.float32, mstype.bfloat16]