Fix typos

This commit is contained in:
RangerUFO 2024-04-10 20:38:18 +08:00
parent 809bd0709d
commit 4e960d67f6
1 changed files with 4 additions and 4 deletions

View File

@ -86,7 +86,7 @@ struct Layer {
static constexpr size_t kGatingEinsumWSize = 2 * kFFHiddenDim * kModelDim; static constexpr size_t kGatingEinsumWSize = 2 * kFFHiddenDim * kModelDim;
static constexpr size_t kConv1dWidth = TConfig::kConv1dWidth; static constexpr size_t kConv1dWidth = TConfig::kConv1dWidth;
static constexpr bool kFFBiases = TConfig::kFFBiases; static constexpr bool kFFBiases = TConfig::kFFBiases;
static constexpr size_t kAOBiaseDim = static constexpr size_t kAOBiasDim =
TConfig::kSoftmaxAttnOutputBiases ? kModelDim : 0; TConfig::kSoftmaxAttnOutputBiases ? kModelDim : 0;
static constexpr size_t kGriffinDim = static constexpr size_t kGriffinDim =
TConfig::kGriffinLayers > 0 ? kModelDim : 0; TConfig::kGriffinLayers > 0 ? kModelDim : 0;
@ -98,7 +98,7 @@ struct Layer {
struct { struct {
ArrayT<float, kAttVecEinsumWSize> attn_vec_einsum_w; ArrayT<float, kAttVecEinsumWSize> attn_vec_einsum_w;
ArrayT<float, kQKVEinsumWSize> qkv_einsum_w; ArrayT<float, kQKVEinsumWSize> qkv_einsum_w;
ArrayT<float, kAOBiaseDim> attention_output_biases; ArrayT<float, kAOBiasDim> attention_output_biases;
}; };
struct { struct {
@ -310,7 +310,7 @@ struct CompressedLayer {
static constexpr size_t kGatingEinsumWSize = TLayer::kGatingEinsumWSize; static constexpr size_t kGatingEinsumWSize = TLayer::kGatingEinsumWSize;
static constexpr size_t kConv1dWidth = TLayer::kConv1dWidth; static constexpr size_t kConv1dWidth = TLayer::kConv1dWidth;
static constexpr bool kFFBiases = TLayer::kFFBiases; static constexpr bool kFFBiases = TLayer::kFFBiases;
static constexpr size_t kAOBiaseDim = TLayer::kAOBiaseDim; static constexpr size_t kAOBiasDim = TLayer::kAOBiasDim;
static constexpr size_t kGriffinDim = TLayer::kGriffinDim; static constexpr size_t kGriffinDim = TLayer::kGriffinDim;
// Compressed Parameters // Compressed Parameters
@ -322,7 +322,7 @@ struct CompressedLayer {
struct { struct {
ArrayT<WeightT, kAttVecEinsumWSize> attn_vec_einsum_w; ArrayT<WeightT, kAttVecEinsumWSize> attn_vec_einsum_w;
ArrayT<WeightT, kQKVEinsumWSize> qkv_einsum_w; ArrayT<WeightT, kQKVEinsumWSize> qkv_einsum_w;
ArrayT<float, kAOBiaseDim> attention_output_biases; ArrayT<float, kAOBiasDim> attention_output_biases;
}; };
struct { struct {