Skip to content

Commit

Permalink
support bfloat16
Browse files Browse the repository at this point in the history
  • Loading branch information
koukairui committed Jan 5, 2024
1 parent d6d2f2f commit f604e54
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mindpet/delta/delta_constants.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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]

0 comments on commit f604e54

Please sign in to comment.