Nick Hill 41deac4a3d [BugFix] 1D query fix for MoE models (#3597)

2024-03-24 16:00:16 -07:00

..

__init__.py

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

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

baichuan.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

bloom.py

[Misc] Fix BLOOM copyright notice (#3591)

2024-03-23 23:30:56 -07:00

chatglm.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

decilm.py

chore(vllm): codespell for spell checking (#2820)

2024-02-21 18:56:01 -08:00

deepseek.py

[BugFix] 1D query fix for MoE models (#3597)

2024-03-24 16:00:16 -07:00

falcon.py

[BugFix] Fix Falcon tied embeddings (#3590)

2024-03-24 06:34:01 -07:00

gemma.py

[BugFix] gemma loading after quantization or LoRA. (#3553)

2024-03-21 13:16:57 -07:00

gpt2.py

[🚀 Ready to be merged] Added support for Jais models (#3183)

2024-03-21 09:45:24 +00:00

gpt_bigcode.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

gpt_j.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

gpt_neox.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

internlm2.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

jais.py

[🚀 Ready to be merged] Added support for Jais models (#3183)

2024-03-21 09:45:24 +00:00

llama.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

mixtral_quant.py

[BugFix] 1D query fix for MoE models (#3597)

2024-03-24 16:00:16 -07:00

mixtral.py

[BugFix] 1D query fix for MoE models (#3597)

2024-03-24 16:00:16 -07:00

mpt.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

olmo.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

opt.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

orion.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

phi.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

qwen2.py

[Bugfix][Model] Fix Qwen2 (#3554)

2024-03-22 00:18:58 +00:00

qwen.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

stablelm.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

starcoder2.py

[Misc] Bump up transformers to v4.39.0 & Remove StarCoder2Config (#3551)

2024-03-21 07:58:12 -07:00