TensorRT-LLMs/tensorrt_llm/_torch/models
William Zhang cdc9e5e645
[None][fix] Properly raise error for nemotron H models (#8697)
Signed-off-by: William Zhang <133824995+2ez4bz@users.noreply.github.com>
2025-10-28 08:59:42 -07:00
..
checkpoints [https://nvbugs/5540138][fix] Fix shape error when duplicating kv. (#8390) 2025-10-17 10:07:29 +08:00
__init__.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
modeling_auto.py [TRTLLM-6746][feat] Enable two-model spec dec for MTP Eagle (#7001) 2025-09-18 12:05:36 -04:00
modeling_bert.py [None][chore] replace print_colored_debug with logger_debug (#8417) 2025-10-22 17:54:38 +08:00
modeling_clip.py [None][feat] Support kv_cahce_reuse for HyperCLOVAX-Vision model (#7789) 2025-10-21 11:11:24 +09:00
modeling_deepseekv3.py [None] [test] Add MNNVL AlltoAll tests to pre-merge (#8601) 2025-10-27 21:39:44 +08:00
modeling_exaone4.py [#6186][feat] Introduce QKNormRoPEAttention module (#6830) 2025-09-05 14:04:41 +02:00
modeling_gemma3.py [#6186][feat] Introduce QKNormRoPEAttention module (#6830) 2025-09-05 14:04:41 +02:00
modeling_gemma3vl.py [TRTLLM-7440][fix] Split fused_input_embed to separate out host sync (#7280) 2025-09-06 23:11:39 -04:00
modeling_gpt_oss.py [None][fix] Fix dummy load format for key models. (#7993) 2025-10-14 11:18:39 +08:00
modeling_hunyuan_dense.py [None][feat] Add Tencent HunYuanDenseV1 model support (#7081) 2025-09-23 09:27:29 +08:00
modeling_hunyuan_moe.py [TRTLLM-7385][feat] Optimize Qwen2/2.5-VL performance (#7250) 2025-09-22 03:40:02 -07:00
modeling_hyperclovax.py [None][feat] Support kv_cahce_reuse for HyperCLOVAX-Vision model (#7789) 2025-10-21 11:11:24 +09:00
modeling_llama_min_latency.py [https://nvbugs/5536131][fix] Fix illegal access issue when scale is not provided in Llama3/4. (#7960) 2025-10-16 22:46:19 +08:00
modeling_llama.py [https://nvbugs/5536131][fix] Fix illegal access issue when scale is not provided in Llama3/4. (#7960) 2025-10-16 22:46:19 +08:00
modeling_llava_next.py [None][feat] Support kv_cahce_reuse for HyperCLOVAX-Vision model (#7789) 2025-10-21 11:11:24 +09:00
modeling_mistral.py [TRTLLM-7410][feat] Enable KV cache reuse and chunked prefill for mistral3.1 (#7628) 2025-09-17 08:11:16 -07:00
modeling_mixtral.py [TRTLLM-7408][feat] Wrap MOE with custom op. (#7277) 2025-09-09 12:18:56 -04:00
modeling_mllama.py feat : support duplicate_kv_weight for qwen3 blockwise scale (#5459) 2025-06-30 11:49:22 +08:00
modeling_multimodal_encoder.py [None][chore] update torch_dtype -> dtype in 'transformers' (#8263) 2025-10-15 17:09:30 +09:00
modeling_multimodal_utils.py [TRTLLM-7328][feat] E-PD Disagg Support via llmapi (3/N) (#7577) 2025-09-22 19:07:18 -07:00
modeling_nanov2vlm.py [TRTLLM-8238][feat] Add EVS support for nano-v2-vlm (#8024) 2025-10-25 05:43:27 -04:00
modeling_nemotron_h.py [None][fix] Properly raise error for nemotron H models (#8697) 2025-10-28 08:59:42 -07:00
modeling_nemotron_nas.py [Perf]: Add residual, norm for nemotron_nas models (#6455) 2025-07-30 09:10:38 -07:00
modeling_nemotron.py feat: Remove not used padding_idx in models (#5385) 2025-06-25 17:19:59 +08:00
modeling_phi3.py [https://nvbugs/5540752][fix] Support quantized Phi4 MM models (#8190) 2025-10-20 06:36:09 -04:00
modeling_phi4mm.py [https://nvbugs/5540752][fix] Support quantized Phi4 MM models (#8190) 2025-10-20 06:36:09 -04:00
modeling_pixtral.py [TRTLLM-7442][model] Remove unnecessary D2H copies (#7273) 2025-09-03 23:14:20 -04:00
modeling_qwen2vl.py [TRTLLM-8737][feat] Support media_io_kwargs on trtllm-serve (#8528) 2025-10-24 12:53:40 -04:00
modeling_qwen3_moe.py [None][fix] Fix dummy load format for key models. (#7993) 2025-10-14 11:18:39 +08:00
modeling_qwen3_next.py [None][chore] Refine qwen3-next implementation. (#8064) 2025-09-30 15:05:13 -04:00
modeling_qwen3.py [None][fix] Fix dummy load format for key models. (#7993) 2025-10-14 11:18:39 +08:00
modeling_qwen_moe.py [TRTLLM-7408][feat] Wrap MOE with custom op. (#7277) 2025-09-09 12:18:56 -04:00
modeling_qwen.py [TRTLLM-7385][feat] Optimize Qwen2/2.5-VL performance (#7250) 2025-09-22 03:40:02 -07:00
modeling_radio.py [TRTLLM-8579][feat] Support quantized model for nano-v2-vlm (#8304) 2025-10-16 09:44:11 +08:00
modeling_seedoss.py [None][feat] Support Seed-OSS model in pytorch backend (#7496) 2025-09-24 03:57:12 -07:00
modeling_siglip.py [None][feat] Support kv_cahce_reuse for HyperCLOVAX-Vision model (#7789) 2025-10-21 11:11:24 +09:00
modeling_speculative.py [None][fix] Eagle: Attention DP (#7939) 2025-10-06 16:52:35 -04:00
modeling_utils.py [None][fix] Fix dummy load format for key models. (#7993) 2025-10-14 11:18:39 +08:00
modeling_vila.py [TRTLLM-8737][feat] Support media_io_kwargs on trtllm-serve (#8528) 2025-10-24 12:53:40 -04:00