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

* Intel Gen 4 Xeon and later support bf16 * fix bf16 notes

Intel Gen 4 Xeon and later support bf16 (#6367)
* Intel Gen 4 Xeon and later support bf16 * fix bf16 notes
f2d51a28 · jiqing-feng · GitHub · 811fd062 · f2d51a28
Unverified Commit f2d51a28 authored Jan 05, 2024 by jiqing-feng Committed by GitHub Jan 05, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/textual_inversion/textual_inversion.py examples/textual_inversion/textual_inversion.py +1 -1

No files found.
--- a/examples/textual_inversion/textual_inversion.py
+++ b/examples/textual_inversion/textual_inversion.py
@@ -341,7 +341,7 @@ def parse_args():
        help=(
            "Whether to use mixed precision. Choose"
            "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(