Skip to content

Commit

Permalink
chore: update following PR
Browse files Browse the repository at this point in the history
  • Loading branch information
NanoCode012 committed Sep 28, 2023
1 parent b29f370 commit de10bd4
Showing 1 changed file with 3 additions and 10 deletions.
13 changes: 3 additions & 10 deletions examples/mistral/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ base_model: mistralai/Mistral-7B-v0.1
base_model_config: mistralai/Mistral-7B-v0.1
model_type: MistralForCausalLM
tokenizer_type: LlamaTokenizer
is_mistral_derived_model: true

load_in_8bit: false
load_in_4bit: false
Expand All @@ -14,18 +15,10 @@ dataset_prepared_path: last_run_prepared
val_set_size: 0.01
output_dir: ./out

sequence_len: 4096
sequence_len: 8192
sample_packing:
pad_to_sequence_len:

adapter:
lora_model_dir:
lora_r: 32
lora_alpha: 16
lora_dropout: 0.05
lora_target_linear: true
lora_fan_in_fan_out:

wandb_project:
wandb_entity:
wandb_watch:
Expand All @@ -51,7 +44,7 @@ resume_from_checkpoint:
local_rank:
logging_steps: 1
xformers_attention:
flash_attention:
flash_attention: true

warmup_steps: 10
eval_steps: 20
Expand Down

0 comments on commit de10bd4

Please sign in to comment.