Skip to content

Commit

Permalink
Merge pull request axolotl-ai-cloud#693 from OpenAccess-AI-Collective…
Browse files Browse the repository at this point in the history
…/update-mistral-example

update mistral lr, sample pack
  • Loading branch information
mhenrichsen committed Oct 7, 2023
2 parents 48cba2e + 8b67313 commit 1853b1f
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions examples/mistral/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,8 @@ val_set_size: 0.01
output_dir: ./out

sequence_len: 8192
sample_packing:
pad_to_sequence_len:
sample_packing: true
pad_to_sequence_len: true

wandb_project:
wandb_entity:
Expand All @@ -30,7 +30,7 @@ micro_batch_size: 2
num_epochs: 3
optimizer: adamw_bnb_8bit
lr_scheduler: cosine
learning_rate: 0.0002
learning_rate: 0.000005

train_on_inputs: false
group_by_length: false
Expand Down Expand Up @@ -59,4 +59,5 @@ fsdp_config:
special_tokens:
bos_token: "<s>"
eos_token: "</s>"

unk_token: "<unk>"

0 comments on commit 1853b1f

Please sign in to comment.