kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
b8818b45be
TensorRT-LLMs/tensorrt_llm/_torch/modules/mamba
History
Luis Vega 0bda1f9780
feat: Nemotron-H model support (#3430) 
* added files for nemotron-h

Signed-off-by: Luis Vega <lvega@nvidia.com>

* use try/except to import RMSNorm

Signed-off-by: Luis Vega <lvega@nvidia.com>

---------

Signed-off-by: Luis Vega <lvega@nvidia.com>
Co-authored-by: QI JUN <22017000+QiJune@users.noreply.github.com>
2025-04-16 14:05:56 -07:00
..
__init__.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
causal_conv1d.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
layernorm_gated.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
mixer.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
selective_state_update.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
softplus.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
ssd_bmm.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
ssd_chunk_scan.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
ssd_chunk_state.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
ssd_combined.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00
ssd_state_passing.py feat: Nemotron-H model support (#3430) 2025-04-16 14:05:56 -07:00