Skip to content

Commit

Permalink
guanaco smollm2
Browse files Browse the repository at this point in the history
  • Loading branch information
abhishekkrthakur committed Nov 4, 2024
1 parent 42d4c12 commit 563aae0
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions configs/llm_finetuning/smollm2_guanaco.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
task: llm-sft
base_model: HuggingFaceTB/SmolLM2-135M-Instruct
project_name: autotrain-smollm2-135m-finetune-guanaco
log: tensorboard
backend: local

data:
path: timdettmers/openassistant-guanaco
train_split: train
valid_split: null
chat_template: null
column_mapping:
text_column: text

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

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

0 comments on commit 563aae0

Please sign in to comment.