Commit ba9ea82

committed

[TRTLLM-13250][feat] Enable MX post-transform Llama receiver

Signed-off-by: Chien-Chun Hung <2679986+chienchunhung@users.noreply.github.com>

1 parent 56aa269 commit ba9ea82Copy full SHA for ba9ea82

6 files changed

tensorrt_llm/_torch
- models/checkpoints/mx
  - checkpoint_loader.py
- pyexecutor
  - model_loader.py
tests/unittest/_torch
- models/checkpoints/mx
  - test_mx_checkpoint_loader.py
- pyexecutor
  - test_model_loader_gms.py
  - test_model_loader_mx.py
- weight_sharing
  - test_mx_source_identity_gate.py

Comments

(0)