Repository navigation
v2.7.1
Summary of Changes
Image: quay.io/modh/fms-hf-tuning:v2.7.1
Mamba Model Support
- Add mamba and cuda dependencies to support tuning Mamba and Jamba models.
Configuration Updates
- New
add_special_tokensargument for special tokens to be added to the tokenizer's vocabulary.
Data Preprocessor Updates
- Enable
streaminginDataSetConfigto load large datasets by utilizing HFIterableDatasets. - Add support for
chat_templateinDataPreProcessorConfig. Fix parsing issue for passingchat_templatevia CLI args. - Allow a tokenizer data handler for users to have direct control over dataset and truncating samples.
- Fix multi-processes broadcast error when running data preprocessor.
- Update data handler backend to support additional types of handlers beyond Map style. Adds support for Filter type data handlers.
- Fix data handler to string conversion bug.
- Allow users to use
toolsanddocumentsin chat templates and pass in additional data arguments to the data handler to enable processing sub field of a data sample. - Add data preprocessing script, decoupling data preprocessor from the tuning. Users can execute the script
offline_data_processing.pyto run data preprocessor standalone.
Dependency Updates
- simpleeval from <1.0 to <2.0
- transformers updated from 4.48.1 to 4.49
- datasets from <3.0 to <4.0
Additional changes
- Support automatic HuggingFace checkpointing for ScatterMoE. Converted checkpoint can be found at
hf_converted_checkpointfolder within every saved checkpoint directory. - Add Github Action to free up disk space.
- Remove deprecated
push_to_hub_tokento resolve warning.
Full list of Changes
- build(deps): Update simpleeval requirement from <1.0,>=0.9.13 to >=0.9.13,<2.0 by @dependabot in #369
- feat: Enable streaming in data preprocessor by @willmj in #437
- feat: Support for add special tokens via cli args by @YashasviChaurasia in #473
- feat: add support for chat template from data config by @YashasviChaurasia in #474
- feat: Add tokenizer data handler. by @dushyantbehl in #487
- build(deps): upgrade transformers to 4.49 by @anhuong in #485
- fix: use main_process_first instead of broadcast_object_list by @willmj in #458
- feat: Update data handler backend and introduce filter based handlers. by @dushyantbehl in #488
- fix: data handler to string conversion bug by @dushyantbehl in #490
- feat: add sum op for trainer controller by @kmehant in #491
- feat: support moe hf chkpt by @kmehant in #486
- chore: Add a GH runner to free up disk space by @aluu317 in #496
- build(deps): Update datasets requirement from <3.0,>=2.15.0 to >=2.15.0,<4.0 by @dependabot in #340
- fix: Remove deprecated push_to_hub_token to resolve warning by @Luka-D in #419
- feat: Add tools and documents usage in chat template by @dushyantbehl in #495
- test:Addition of data preprocessing script, decoupling data preprocessor from the tuning by @Abhishek-TAMU in #459
- fix: bug which caused test case failures after PR merge by @dushyantbehl in #500
- build(deps): changes needed to support mamba/jamba model by @anhuong in #400
Full Changelog: v2.6.0...v2.7.1