Skip to content

Commit

Permalink
add config llama 1b
Browse files Browse the repository at this point in the history
  • Loading branch information
abhishekkrthakur committed Sep 26, 2024
1 parent 2f50992 commit 1d1ce2b
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions configs/llm_finetuning/llama32-1b-sft.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
task: llm-sft
base_model: meta-llama/Llama-3.2-1B
project_name: autotrain-llama32-1b-finetune
log: tensorboard
backend: local

data:
path: HuggingFaceH4/no_robots
train_split: train
valid_split: null
chat_template: tokenizer
column_mapping:
text_column: messages

params:
block_size: 2048
model_max_length: 4096
epochs: 2
batch_size: 1
lr: 1e-5
peft: true
quantization: int4
target_modules: all-linear
padding: right
optimizer: paged_adamw_8bit
scheduler: cosine
gradient_accumulation: 8
mixed_precision: bf16
merge_adapter: true

hub:
username: ${HF_USERNAME}
token: ${HF_TOKEN}
push_to_hub: true

0 comments on commit 1d1ce2b

Please sign in to comment.