TensorRT-LLMs/tests/unittest/_torch/modeling
2ez4bz ab7434ac62
[feat] Enable TP and batching for PixtralVisionModel / Mistral3VLM (#6152)
Signed-off-by: William Zhang <133824995+2ez4bz@users.noreply.github.com>
2025-07-22 11:06:41 -07:00
..
test_modeling_bert.py feat: no-cache attention in PyTorch workflow (#3085) 2025-04-05 01:54:32 +08:00
test_modeling_clip.py feat: add Pytorch support of Vision Encoder for multimodal models (#3791) 2025-05-03 05:13:47 +08:00
test_modeling_deepseek.py [TRTLLM-5530][BREAKING CHANGE] refactor: unify KvCacheConfig in LLM class for pytorch backend (#5752) 2025-07-16 16:42:59 +08:00
test_modeling_gemma3.py fix: Flush stale PlanParams with custom attention mask (#6163) 2025-07-21 09:55:09 +08:00
test_modeling_llama_min_latency.py [TRTLLM-5493] Add core infrastructure to enable loading of custom checkpoint formats (#5372) 2025-07-17 00:50:30 +08:00
test_modeling_llama.py [fix] speedup modeling unittests (#5579) 2025-06-30 06:30:45 +03:00
test_modeling_mistral.py [fix] Fix mistral unit tests due to transformers upgrade (#5904) 2025-07-10 10:45:27 -07:00
test_modeling_mixtral.py [TRTLLM-5493] Add core infrastructure to enable loading of custom checkpoint formats (#5372) 2025-07-17 00:50:30 +08:00
test_modeling_mllama.py Update transformers to 4.53.0 (#5747) 2025-07-09 09:32:24 -07:00
test_modeling_nemotron_h.py [Infra] - Waive failed cases on recent post-merge (#6212) 2025-07-21 21:00:18 +08:00
test_modeling_nemotron_nas.py [fix][test] Speedup Nemotron NAS unittests (#5202) 2025-06-15 11:26:03 +03:00
test_modeling_nemotron.py [fix] speedup modeling unittests (#5579) 2025-06-30 06:30:45 +03:00
test_modeling_out_of_tree.py chores: merge examples for v1.0 doc (#5736) 2025-07-08 21:00:42 -07:00
test_modeling_phi3.py feat: TRTLLM-5574 Add phi-4-multimodal pytorch-backend support (#5644) 2025-07-17 06:30:58 +08:00
test_modeling_pixtral.py [feat] Enable TP and batching for PixtralVisionModel / Mistral3VLM (#6152) 2025-07-22 11:06:41 -07:00
test_modeling_qwen_moe.py [TRTLLM-5493] Add core infrastructure to enable loading of custom checkpoint formats (#5372) 2025-07-17 00:50:30 +08:00
test_modeling_qwen.py feat: no-cache attention in PyTorch workflow (#3085) 2025-04-05 01:54:32 +08:00
test_modeling_siglip.py feat: Update Gemma3 Vision Encoder (#5973) 2025-07-14 22:38:10 +08:00
test_modeling_vila.py feat: llama4 input processor (#3383) 2025-04-25 16:47:14 -07:00