huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use some consts for ggml values. (#312)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-02 22:03:05 +01:00
committed by GitHub
parent e635f18eda
commit 634700d84a
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
candle-core/src/ggml.rs
View File

@ -3,6 +3,10 @@
use crate::Result; use crate::Result;
use byteorder::{LittleEndian, ReadBytesExt}; use byteorder::{LittleEndian, ReadBytesExt};
// Default to QK_K 256 rather than 64.
pub const QK_K: usize = 256;
pub const K_SCALE_SIZE: usize = 12;
// https://github.com/ggerganov/llama.cpp/blob/468ea24fb4633a0d681f7ac84089566c1c6190cb/llama.h#L37 // https://github.com/ggerganov/llama.cpp/blob/468ea24fb4633a0d681f7ac84089566c1c6190cb/llama.h#L37
#[derive(Debug, Clone, Copy, PartialEq, Eq)] #[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum Magic { enum Magic {
@ -164,12 +168,12 @@ impl GgmlDType {
// https://github.com/ggerganov/llama.cpp/blob/468ea24fb4633a0d681f7ac84089566c1c6190cb/ggml.c#L932 // https://github.com/ggerganov/llama.cpp/blob/468ea24fb4633a0d681f7ac84089566c1c6190cb/ggml.c#L932
Self::Q8_0 => 34, Self::Q8_0 => 34,
Self::Q8_1 => 36, Self::Q8_1 => 36,
Self::Q2K => 256 / 16 + 256 / 4 + 2 * 2, Self::Q2K => QK_K / 16 + QK_K / 4 + 2 * 2,
Self::Q3K => 256 / 8 + 256 / 4 + 12 + 2, Self::Q3K => QK_K / 8 + QK_K / 4 + 12 + 2,
// https://github.com/ggerganov/llama.cpp/blob/468ea24fb4633a0d681f7ac84089566c1c6190cb/k_quants.h#L82 // https://github.com/ggerganov/llama.cpp/blob/468ea24fb4633a0d681f7ac84089566c1c6190cb/k_quants.h#L82
Self::Q4K => 256 / 2 + 12 + 2 * 2, Self::Q4K => QK_K / 2 + K_SCALE_SIZE + 2 * 2,
Self::Q5K => 256 / 8 + 256 / 2 + 2 * 2 + 12, Self::Q5K => QK_K / 8 + QK_K / 2 + 2 * 2 + K_SCALE_SIZE,
Self::Q6K => 3 * 256 / 4 + 256 / 16 + 2, Self::Q6K => 3 * QK_K / 4 + QK_K / 16 + 2,
} }
} }
@ -183,12 +187,7 @@ impl GgmlDType {
Self::Q5_1 => 32, Self::Q5_1 => 32,
Self::Q8_0 => 32, Self::Q8_0 => 32,
Self::Q8_1 => 32, Self::Q8_1 => 32,
// Default to QK_K 256 rather than 64. Self::Q2K | Self::Q3K | Self::Q4K | Self::Q5K | Self::Q6K => QK_K,
Self::Q2K => 256,
Self::Q3K => 256,
Self::Q4K => 256,
Self::Q5K => 256,
Self::Q6K => 256,
} }
} }
} }