Wing Lian 229b9165aa fix test and pylint checks

2023-08-07 09:38:05 -04:00

..

fixtures

experimental llama 2 chat support (#296)

2023-08-06 17:40:52 -04:00

monkeypatch

add unit tests for cum seq lens, add ability to build cu_seq_lens from positional ids, fix prompt test

2023-08-07 09:38:04 -04:00

test_dict.py

Ignore unsupported-binary-operation

2023-05-31 02:53:53 +09:00

test_expand_mask.py

fix flash-attn, xformers, packing, support chatml

2023-08-07 09:38:04 -04:00

test_packed_dataset.py

set position ids and use block diagonal attn mask

2023-08-07 09:38:03 -04:00

test_prompt_tokenizers.py

add unit tests for cum seq lens, add ability to build cu_seq_lens from positional ids, fix prompt test

2023-08-07 09:38:04 -04:00

test_prompters.py

fix test and pylint checks

2023-08-07 09:38:05 -04:00

test_tokenizers.py

optionally define whether to use_fast tokenizer

2023-06-25 10:19:49 -04:00

test_validation.py

validation for sample packing and doc

2023-08-07 09:38:03 -04:00