Skip to content

Commit

Permalink
fix linting
Browse files Browse the repository at this point in the history
  • Loading branch information
lvhan028 committed Nov 1, 2024
1 parent 5125e87 commit 4532718
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/turbomind/models/llama/LlamaDecoderLayerWeight.cc
Original file line number Diff line number Diff line change
Expand Up @@ -400,7 +400,8 @@ void LlamaDecoderLayerWeight<T>::loadModel(std::string dir_path, FtCudaDataType
loadWeights(
ffn_weights.intermediate, dir_path + ".feed_forward.w3", tensor_para_rank_, type, tensor_para_size_);
loadWeights(ffn_weights.output, dir_path + ".feed_forward.w2", tensor_para_rank_, type, tensor_para_size_);
} else {
}
else {
loadWeights(moe_weights.gate, dir_path + ".moe_ffn.gate", type);
for (size_t i = 0; i < moe_weights.experts.size(); ++i) {
std::string weight_name = dir_path + ".moe_ffn.experts." + std::to_string(i);
Expand Down

0 comments on commit 4532718

Please sign in to comment.