Murali Andoorveedu fc912e0886 [Models] Support Qwen model with PP (#6974) ...

Signed-off-by: Muralidhar Andoorveedu <muralidhar.andoorveedu@centml.ai>

2024-08-01 12:40:43 -07:00

..

__init__.py

[Model] Initialize support for InternVL2 series models (#6514)

2024-07-29 10:16:30 +00:00

arctic.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

baichuan.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

blip2.py

[Model] Initial support for BLIP-2 (#5920)

2024-07-27 11:53:07 +00:00

blip.py

[Model] Initial support for BLIP-2 (#5920)

2024-07-27 11:53:07 +00:00

bloom.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

chameleon.py

[Bugfix][Model] Skip loading lm_head weights if using tie_word_embeddings (#6758)

2024-07-31 19:49:11 -07:00

chatglm.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

clip.py

[VLM] Minor space optimization for ClipVisionModel (#6436)

2024-07-15 17:29:51 +08:00

commandr.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

dbrx.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

decilm.py

[Model] Add base class for LoRA-supported models (#5018)

2024-06-27 16:03:04 +08:00

deepseek_v2.py

[Model] Pipeline Parallel Support for DeepSeek v2 (#6519)

2024-07-23 12:22:09 -07:00

deepseek.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

falcon.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

fuyu.py

[Bugfix] Fix feature size calculation for LLaVA-NeXT (#6982)

2024-07-31 23:46:17 +08:00

gemma2.py

[Bugfix] Lower gemma's unloaded_params exception to warning (#7002)

2024-08-01 12:01:07 -07:00

gemma.py

[Bugfix] Lower gemma's unloaded_params exception to warning (#7002)

2024-08-01 12:01:07 -07:00

gpt2.py

[ Misc ] non-uniform quantization via compressed-tensors for Llama (#6515)

2024-07-18 22:39:18 -04:00

gpt_bigcode.py

[Bugfix] GPTBigCodeForCausalLM: Remove lm_head from supported_lora_modules. (#6326)

2024-07-11 07:05:59 -07:00

gpt_j.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

gpt_neox.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

interfaces.py

[BugFix][Model] Jamba - Handle aborted requests, Add tests and fix cleanup bug (#6425)

2024-07-16 01:32:55 +00:00

intern_vit.py

[Bugfix] Fix RMSNorm forward in InternViT attention qk_layernorm (#6992)

2024-08-01 12:00:28 -07:00

internlm2.py

[Model] Initialize support for InternVL2 series models (#6514)

2024-07-29 10:16:30 +00:00

internvl.py

[Bugfix] Fix feature size calculation for LLaVA-NeXT (#6982)

2024-07-31 23:46:17 +08:00

jais.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

jamba.py

[Model] use FusedMoE layer in Jamba (#6935)

2024-07-31 12:00:24 -07:00

llama_embedding.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

llama.py

[Bugfix][Model] Skip loading lm_head weights if using tie_word_embeddings (#6758)

2024-07-31 19:49:11 -07:00

llava_next.py

[Bugfix] Fix feature size calculation for LLaVA-NeXT (#6982)

2024-07-31 23:46:17 +08:00

llava.py

[Bugfix] Fix vocab_size field access in llava_next.py (#6624)

2024-07-22 05:02:51 +00:00

medusa.py

[Speculative Decoding] Medusa Implementation with Top-1 proposer (#4978)

2024-07-09 18:34:02 -07:00

minicpm.py

[Bugfix][Model] Skip loading lm_head weights if using tie_word_embeddings (#6758)

2024-07-31 19:49:11 -07:00

minicpmv.py

[Bugfix] Clean up MiniCPM-V (#6939)

2024-07-31 14:39:19 +00:00

mixtral_quant.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

mixtral.py

[ Misc ] non-uniform quantization via compressed-tensors for Llama (#6515)

2024-07-18 22:39:18 -04:00

mlp_speculator.py

[Bugfix] MLPSpeculator: Use ParallelLMHead in tie_weights=False case. (#6303)

2024-07-10 09:04:07 -04:00

mpt.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

na_vit.py

[Bugfix] Clean up MiniCPM-V (#6939)

2024-07-31 14:39:19 +00:00

nemotron.py

[Model] Support Nemotron models (Nemotron-3, Nemotron-4, Minitron) (#6611)

2024-07-26 14:33:42 -04:00

olmo.py

[Bugfix][Model] Skip loading lm_head weights if using tie_word_embeddings (#6758)

2024-07-31 19:49:11 -07:00

opt.py

[Model] Initial support for BLIP-2 (#5920)

2024-07-27 11:53:07 +00:00

orion.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

paligemma.py

[Bugfix] Lower gemma's unloaded_params exception to warning (#7002)

2024-08-01 12:01:07 -07:00

persimmon.py

[Model] Initialize Fuyu-8B support (#3924)

2024-07-14 05:27:14 +00:00

phi3_small.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

phi3v.py

[Bugfix] Fix feature size calculation for LLaVA-NeXT (#6982)

2024-07-31 23:46:17 +08:00

phi.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

qwen2_moe.py

[Model] Pipeline parallel support for Qwen2 (#6924)

2024-07-31 18:49:51 -07:00

qwen2.py

[Model] Pipeline parallel support for Qwen2 (#6924)

2024-07-31 18:49:51 -07:00

qwen.py

[Models] Support Qwen model with PP (#6974)

2024-08-01 12:40:43 -07:00

stablelm.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

starcoder2.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

utils.py

[Bugfix] Support cpu offloading with fp8 quantization (#6960)

2024-07-31 12:47:46 -07:00

xverse.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00