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

…621)
huggingface · Jul 17, 2023 · bc28732 · bc28732
1 parent a2cf1bd
commit bc28732
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,
 }