Skip to content

GitLab

Unverified Commit f2d51a28 authored by jiqing-feng's avatar jiqing-feng Committed by GitHub
Browse files

Intel Gen 4 Xeon and later support bf16 (#6367) 

* Intel Gen 4 Xeon and later support bf16

* fix bf16 notes
parent 811fd062
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/textual_inversion/textual_inversion.py
View file @ f2d51a28
...@@ -341,7 +341,7 @@ def parse_args(): ...@@ -341,7 +341,7 @@ def parse_args():
help=( help=(
"Whether to use mixed precision. Choose" "Whether to use mixed precision. Choose"
"between fp16 and bf16 (bfloat16). Bf16 requires PyTorch >= 1.10." "between fp16 and bf16 (bfloat16). Bf16 requires PyTorch >= 1.10."
"and an Nvidia Ampere GPU." "and Nvidia Ampere GPU or Intel Gen 4 Xeon (and later) ."
), ),
) )
parser.add_argument( parser.add_argument(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment