diff --git a/candle-core/src/safetensors.rs b/candle-core/src/safetensors.rs index 22f8286f5..162928ec7 100644 --- a/candle-core/src/safetensors.rs +++ b/candle-core/src/safetensors.rs @@ -206,7 +206,7 @@ fn convert(view: &st::TensorView<'_>, device: &Device) -> Result { convert_with_cast_::(view, device, conv) } st::Dtype::U32 => convert_::(view, device), - st::Dtype::I32 => onvert_::(view, device), + st::Dtype::I32 => convert_::(view, device), st::Dtype::I64 => convert_::(view, device), st::Dtype::BF16 => convert_::(view, device), st::Dtype::F16 => convert_::(view, device), diff --git a/candle-kernels/src/indexing.cu b/candle-kernels/src/indexing.cu index e60ef5a95..2f3df4de1 100644 --- a/candle-kernels/src/indexing.cu +++ b/candle-kernels/src/indexing.cu @@ -199,7 +199,7 @@ IS_OP(int64_t, int64_t, is_i64_i64) IS_OP(int32_t, int64_t, is_i64_i32) IS_OP(float, uint32_t, is_u32_f32) -IS_OP(double, uint32_t, is_u32_f64)w +IS_OP(double, uint32_t, is_u32_f64) IS_OP(uint8_t, uint32_t, is_u32_u8) IS_OP(int32_t, uint32_t, is_u32_i32) IS_OP(int64_t, uint32_t, is_u32_i64)