Wing Lian 1a538be9c2 add a prelim test for expading the 4d mask

2024-01-26 00:41:24 -05:00

..

core

add gptneox embeddings, fix phi2 inputs, also fix the casting (#1083)

2024-01-10 22:32:43 -05:00

e2e

Phi2 multipack (#1173)

2024-01-23 12:54:36 -05:00

fixtures

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

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

monkeypatch

Multipack simplify for Mixtral (#1142)

2024-01-18 16:23:49 -05:00

prompt_strategies

Feat/chatml add system message (#1117)

2024-01-25 08:24:27 +01: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_dict.py

fix DefaultDict.__or__

2023-08-13 01:15:50 +00:00

test_expand_mask.py

add a prelim test for expading the 4d mask

2024-01-26 00:41:24 -05:00

test_normalize_config.py

set fp16 to false if bf16, update bf16: auto in example YAMLs (#1122) [skip ci]

2024-01-22 18:44:01 -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

streaming multipack for pretraining dataset (#959)

2024-01-05 22:13:21 -05:00

test_prompt_tokenizers.py

fix mistral prompt assembly (#982)

2023-12-21 08:00:55 -08:00

test_prompters.py

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

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

test_tokenizers.py

Feat: Warns to add to modules_to_save when adding tokens or switching special_tokens (#787)

2023-12-22 21:49:07 +09:00

test_validation.py

Phi2 multipack (#1173)

2024-01-23 12:54:36 -05:00