don't fail if can't cast weights due to offload when merging (#1172) [skip ci] fb7f9b9 unverified winglian commited on Jan 23
support for explicit test_dataset definition for evals (#786) cda52dc unverified winglian commited on Jan 23
improve vram use w gradient checkpointing (#1167) [skip ci] 802f966 unverified winglian commited on Jan 23
Add mlflow callback for pushing config to mlflow artifacts (#1125) b8e5603 unverified JohanWork commited on Jan 22
set fp16 to false if bf16, update bf16: auto in example YAMLs (#1122) [skip ci] 782b6a4 unverified winglian Nanobit commited on Jan 22
make sure the model config loader respects the model_revision too (#1160) [skip-ci] fccb542 unverified winglian commited on Jan 22
feat(dataset): add config to keep processed dataset in memory (#1152) 3db5f2f unverified Nanobit commited on Jan 20
Add shifted sparse attention (#973) [skip-ci] 1d70f24 unverified jrc joecummings winglian commited on Jan 18
fix(preprocess): Make sure dataset not loaded from cache when using preprocess cli (#1136) 1e56b88 unverified Nanobit commited on Jan 17
Enable or disable bf16 support based on availability (#1116) 0865613 unverified Simon Hällqvist commited on Jan 14
Disable caching on `--disable_caching` in CLI (#1110) d66b101 unverified casperhansen winglian commited on Jan 13
add gptneox embeddings, fix phi2 inputs, also fix the casting (#1083) 78c5b19 unverified winglian commited on Jan 11
optimize calculation of cu_seqlens from position_ids (#1084) [skip ci] 90036eb unverified winglian commited on Jan 10
additional logging to get maximum token length of a sequence in the dataset (#1066) [skip ci] 2f2582e unverified winglian commited on Jan 10
update sharegpt conversations when chatml chat template is set (#1075) [skip ci] 0ce1a65 unverified winglian commited on Jan 10
fix: `train_on_inputs: true` ignored for sharegpt (#1045) [skip ci] 043c386 unverified Nanobit winglian commited on Jan 10
be more robust about checking embedding modules for lora finetunes (#1074) [skip ci] 0f10080 unverified winglian commited on Jan 10
swap the data collator for evals if not using sample packing (#1076) ead34c5 unverified winglian commited on Jan 10
Add: mlflow for experiment tracking (#1059) [skip ci] 090c24d unverified Johan Hansson winglian commited on Jan 9
Cosine learning rate schedule - minimum learning rate (#1062) 04b978b unverified ricdomolm winglian commited on Jan 9
Efficiently get the length of the tokenized docs (#1063) 81d3845 unverified ricdomolm winglian commited on Jan 8
streaming multipack for pretraining dataset (#959) 553c80f unverified jinwonkim93 jinwonkim93@github.com winglian commited on Jan 6
feat: always push checkpoint to hub if set (#1049) [skip ci] cbdbf9e unverified Nanobit commited on Jan 5
feature: better device mapping for large models (#918) bdfefaf unverified dg-kalle Karl-Johan Alm winglian commited on Jan 5
Added chatglm3 conversation type for training models like TinyLLama (#1036) 59b2d30 unverified xaviviro commited on Jan 4
bump transformers and update attention class map name (#1023) bcc78d8 unverified winglian commited on Jan 3