Skip to content

Commit

Permalink
new config
Browse files Browse the repository at this point in the history
  • Loading branch information
ahmeda14960 committed Sep 11, 2024
1 parent 3fe04f8 commit c6e64aa
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions config/olmo_sft.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,8 @@ model: # 7B class model
# use_flash_attention: True
# use_bias: false
# use_layer_norm_weight: false
initialize_from_hf: "allenai/OLMo-7B-0724-hf"
use_hf_model_config: true
initialize_from_hf: "allenai/OLMo-7B-0724-hf"
use_hf_model_config: true
#flash_attention_block_size: 1024
trainer:
tracker:
Expand Down

0 comments on commit c6e64aa

Please sign in to comment.