TensorRT-LLMs/tests/unittest/_torch/modeling
nv-guomingz 4e4d18826f
chore: [Breaking Change] Rename cuda_graph_config padding_enabled fie… (#6003)
Signed-off-by: nv-guomingz <137257613+nv-guomingz@users.noreply.github.com>
2025-07-15 15:50:03 +09: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 chore: [Breaking Change] Rename cuda_graph_config padding_enabled fie… (#6003) 2025-07-15 15:50:03 +09:00
test_modeling_gemma3.py enh: Bidirectional mask with multiple images for Gemma3 (#5976) 2025-07-14 22:39:18 +08:00
test_modeling_llama_min_latency.py [fix] speedup modeling unittests (#5579) 2025-06-30 06:30:45 +03: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 [fix] speedup modeling unittests (#5579) 2025-06-30 06:30:45 +03: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 [TRTLLM-4923][feat] Enable CUDA graphs for Nemotron-H (#5646) 2025-07-03 11:07:51 +03: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_pixtral.py feat(models): Mistral3.1 VLM pytorch backend support (#5529) 2025-07-09 13:17:40 -07:00
test_modeling_qwen_moe.py feat: add qwen2 moe to torch flow; fix wrong imported KvCacheConfig in gpqa… (#3369) 2025-04-10 22:45:57 +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