Maxime fac2d98c26 Add MPS support (#1264) ...

* add mps support

* linter stuff

* CI fixes

* install packaging for various tests

* Update setup.py

* Revert "install packaging for various tests"

This reverts commit 980e7aa44d.

* Revert "CI fixes"

This reverts commit 4609e3b166.

---------

Co-authored-by: Wing Lian <wing.lian@gmail.com>

2024-02-12 08:30:32 -05:00

..

cerebras

Update qlora.yml - remove max_packed_sequence_len (#1210) [skip ci]

2024-01-26 07:43:05 -05:00

code-llama

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

2024-01-22 18:44:01 -05:00

colab-notebooks

lock pytorch (#1247) [skip ci]

2024-02-06 07:48:26 -05:00

falcon

Falcon embeddings (#1149) [skip docker]

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

gptj

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

2024-01-22 18:44:01 -05:00

jeopardy-bot

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

2024-01-22 18:44:01 -05:00

llama-2

Peft lotfq (#1222)

2024-01-28 18:50:08 -05:00

mamba

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

2024-01-22 18:44:01 -05:00

mistral

Mixtral fixes 20240124 (#1192) [skip ci]

2024-01-24 14:59:57 -05:00

mpt-7b

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

2024-01-22 18:44:01 -05:00

openllama-3b

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

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

phi

Mixtral fixes 20240124 (#1192) [skip ci]

2024-01-24 14:59:57 -05:00

pythia

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

2024-01-22 18:44:01 -05:00

pythia-12b

Feat(wandb): Refactor to be more flexible (#767)

2023-12-04 22:17:25 +09:00

qwen

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

2024-01-22 18:44:01 -05:00

redpajama

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

2024-01-22 18:44:01 -05:00

replit-3b

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

2024-01-22 18:44:01 -05:00

tiny-llama

Add MPS support (#1264)

2024-02-12 08:30:32 -05:00

xgen-7b

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

2024-01-22 18:44:01 -05:00

yi-34B-chat

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

2024-01-22 18:44:01 -05:00