yuehuayingxueluo fa4fbdbffb adapted to pad_context_forward

2024-01-11 13:44:06 +00:00

..

amp

[npu] add npu support for hybrid plugin and llama (#5090)

2023-11-22 19:23:21 +08:00

builder

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

communication

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

context

[hotfix] fix torch 2.0 compatibility (#4936)

2023-10-18 11:05:25 +08:00

engine

[npu] add npu support for gemini and zero (#5067)

2023-11-20 16:12:41 +08:00

inference

adapted to pad_context_forward

2024-01-11 13:44:06 +00:00

nn

[npu] add npu support for gemini and zero (#5067)

2023-11-20 16:12:41 +08:00

pipeline

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

registry

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

tensor

[hotfix] fix torch 2.0 compatibility (#4936)

2023-10-18 11:05:25 +08:00

trainer

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

utils

[npu] add npu support for hybrid plugin and llama (#5090)

2023-11-22 19:23:21 +08:00

zero

[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088)

2023-11-28 16:54:42 +08:00

__init__.py

[bug] fix get_default_parser in examples (#4764)

2023-09-21 10:42:25 +08:00

constants.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

core.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

global_variables.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

initialize.py

[moe] merge moe into main (#4978)

2023-11-02 02:21:24 +00:00