Skip to content

Commit 9588f19

Browse files
committed
train : change default FA argument (#7528)
1 parent 3cbd23e commit 9588f19

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

common/train.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1052,7 +1052,7 @@ struct train_params_common get_default_train_params_common() {
10521052

10531053
params.custom_n_ctx = false;
10541054

1055-
params.use_flash = true;
1055+
params.use_flash = false;
10561056
params.use_checkpointing = true;
10571057

10581058
params.sample_start = "";

examples/convert-llama2c-to-ggml/convert-llama2c-to-ggml.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -774,7 +774,7 @@ static struct train_params get_default_train_params() {
774774

775775
params.samples_start_after_nl = false;
776776
params.use_adam = true;
777-
params.use_flash = true;
777+
params.use_flash = false;
778778
params.use_scratch = true;
779779

780780
// only adam

0 commit comments

Comments
 (0)