fix(launcher): Rename `b-float16` to `bfloat16` in the launcher arg (#621)

bc287324 · Nicolas Patry · GitHub · a2cf1bdb · bc287324
Unverified Commit bc287324 authored Jul 17, 2023 by Nicolas Patry Committed by GitHub Jul 17, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

launcher/src/main.rs launcher/src/main.rs +1 -0

No files found.
--- a/launcher/src/main.rs
+++ b/launcher/src/main.rs
@@ -41,6 +41,7 @@ impl std::fmt::Display for Quantization {
 #[derive(Clone, Copy, Debug, ValueEnum)]
 enum Dtype {
    Float16,
+    #[clap(name = "bfloat16")]
    BFloat16,
 }