File tree Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -203,20 +203,20 @@ void LlamaNode::load_params(struct gpt_params ¶ms) {
203
203
204
204
// split mode
205
205
if (split_mode == " none" ) {
206
- params.split_mode = LLAMA_SPLIT_NONE ;
206
+ params.split_mode = LLAMA_SPLIT_MODE_NONE ;
207
207
} else if (split_mode == " layer" ) {
208
- params.split_mode = LLAMA_SPLIT_LAYER ;
208
+ params.split_mode = LLAMA_SPLIT_MODE_LAYER ;
209
209
} else if (split_mode == " row" ) {
210
- params.split_mode = LLAMA_SPLIT_ROW ;
210
+ params.split_mode = LLAMA_SPLIT_MODE_ROW ;
211
211
}
212
212
213
213
// rope_scaling_type
214
214
if (rope_scaling_type == " none" ) {
215
- params.rope_scaling_type = LLAMA_ROPE_SCALING_NONE ;
215
+ params.rope_scaling_type = LLAMA_ROPE_SCALING_TYPE_NONE ;
216
216
} else if (rope_scaling_type == " linear" ) {
217
- params.rope_scaling_type = LLAMA_ROPE_SCALING_LINEAR ;
217
+ params.rope_scaling_type = LLAMA_ROPE_SCALING_TYPE_LINEAR ;
218
218
} else if (rope_scaling_type == " yarn" ) {
219
- params.rope_scaling_type = LLAMA_ROPE_SCALING_YARN ;
219
+ params.rope_scaling_type = LLAMA_ROPE_SCALING_TYPE_YARN ;
220
220
}
221
221
222
222
// numa
You can’t perform that action at this time.
0 commit comments