wenyujin333 d6ea427f04

[Model] Add support for Qwen2MoeModel (#3346)

2024-03-28 15:19:59 +00:00

..

layers

[Kernel] Add Triton MoE kernel configs for DBRX on A100 (#3679)

2024-03-27 22:22:25 -07:00

models

[Model] Add support for Qwen2MoeModel (#3346)

2024-03-28 15:19:59 +00:00

parallel_utils

[Bugfix] [Hotfix] fix nccl library name (#3661)

2024-03-27 17:23:54 +00:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

guided_decoding.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

guided_logits_processors.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

model_loader.py

[Feature] Add vision language model support. (#3042)

2024-03-25 14:16:30 -07:00

neuron_model_loader.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

sampling_metadata.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

utils.py

[Hardware][Neuron] Refactor neuron support (#3471)

2024-03-22 01:22:17 +00:00

weight_utils.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00