NanoCode012 bf4cd67252 feat: validate sample packing requires flash_attention (#1465) ...

* feat: validate sample packing requires flash_attention

* fix: check for sdp_attn per suggestion

* feat: add FA to tests

2024-04-05 12:47:32 +09:00

..

core

fix(dataset): normalize tokenizer config and change hash from tokenizer class to tokenizer path (#1298)

2024-03-25 15:34:54 +09:00

e2e

strip out hacky qlora-fsdp workarounds now that qlora-fsdp fixes are upstreamed (#1428)

2024-03-21 11:56:13 -04:00

fixtures

Respect sequence_len in config for type: llama2_chat (#926)

2023-12-12 09:39:22 -08:00

monkeypatch

support for true batches with multipack (#1230)

2024-02-01 10:18:42 -05:00

prompt_strategies

Feat: Add sharegpt multirole (#1137)

2024-03-19 20:51:49 +09:00

utils

Add shifted sparse attention (#973) [skip-ci]

2024-01-18 10:16:07 -05:00

test_data.py

Fix pretraining with iterable/streaming Dataset (#556)

2023-09-13 00:16:40 -04:00

test_datasets.py

Support loading datasets saved via save_to_disk (#1432)

2024-03-29 00:19:36 -04:00

test_dict.py

Pydantic 2.x cfg (#1239)

2024-02-26 12:24:14 -05:00

test_expand_mask.py

Attention mask and position id fixes for packing (#285)

2023-08-12 15:14:56 -04:00

test_freeze.py

Train parameters exclusively in specific ranges (#1390)

2024-03-14 11:05:42 -04:00

test_normalize_config.py

more fixes 20240228 (#1342) [skip ci]

2024-02-28 12:57:45 -05:00

test_packed_batch_sampler.py

support for true batches with multipack (#1230)

2024-02-01 10:18:42 -05:00

test_packed_dataset.py

Attention mask and position id fixes for packing (#285)

2023-08-12 15:14:56 -04:00

test_packed_pretraining.py

Pretrain transforms (#1261)

2024-02-06 00:37:03 -05:00

test_prompt_tokenizers.py

ORPO (#1419)

2024-03-18 13:10:00 -04:00

test_prompters.py

Attention mask and position id fixes for packing (#285)

2023-08-12 15:14:56 -04:00

test_schedulers.py

Scheduler implementation of Continual Pre-Training of Large Language Models: How to (re)warm your model? (#1273)

2024-02-12 21:23:28 -08:00

test_tokenizers.py

Support for additional_special_tokens (#1221) [skip ci]

2024-01-31 18:13:13 -05:00

test_validation.py

feat: validate sample packing requires flash_attention (#1465)

2024-04-05 12:47:32 +09:00