huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

s/panic/bail/

Browse Source
This commit is contained in:
Nicolas Patry
2023-08-25 18:05:07 +02:00
parent 4826a4212e
commit c105550405
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/llama/main.rs
View File

@ -12,7 +12,7 @@ extern crate accelerate_src;
#[cfg(feature = "mkl")] #[cfg(feature = "mkl")]
extern crate intel_mkl_src; extern crate intel_mkl_src;
use anyhow::{Error as E, Result}; use anyhow::{bail, Error as E, Result};
use clap::Parser; use clap::Parser;
use candle::{DType, Tensor}; use candle::{DType, Tensor};
@ -104,7 +104,7 @@ fn main() -> Result<()> {
Some("f16") => DType::F16, Some("f16") => DType::F16,
Some("bf16") => DType::BF16, Some("bf16") => DType::BF16,
Some("f32") => DType::F32, Some("f32") => DType::F32,
Some(dtype) => panic!("Unsupported dtype {dtype}"), Some(dtype) => bail!("Unsupported dtype {dtype}"),
None => DType::F16, None => DType::F16,
}; };
let (llama, tokenizer_filename, cache) = match args.npy { let (llama, tokenizer_filename, cache) = match args.npy {

4
candle-examples/examples/llama_multiprocess/main.rs
View File

@ -9,7 +9,7 @@
#[cfg(feature = "mkl")] #[cfg(feature = "mkl")]
extern crate intel_mkl_src; extern crate intel_mkl_src;
use anyhow::{Error as E, Result}; use anyhow::{bail, Error as E, Result};
use clap::Parser; use clap::Parser;
use candle::{DType, Device, Tensor}; use candle::{DType, Device, Tensor};
@ -125,7 +125,7 @@ fn main() -> Result<()> {
Some("f16") => DType::F16, Some("f16") => DType::F16,
Some("bf16") => DType::BF16, Some("bf16") => DType::BF16,
Some("f32") => DType::F32, Some("f32") => DType::F32,
Some(dtype) => panic!("Unsupported dtype {dtype}"), Some(dtype) => bail!("Unsupported dtype {dtype}"),
None => DType::F16, None => DType::F16,
}; };