diff --git a/tensorrt_llm/llmapi/llm_args.py b/tensorrt_llm/llmapi/llm_args.py index 1caeff83a22c..e078c1b29181 100644 --- a/tensorrt_llm/llmapi/llm_args.py +++ b/tensorrt_llm/llmapi/llm_args.py @@ -1132,6 +1132,12 @@ def __init__(self, **kwargs): self.max_total_draft_tokens = kwargs[ 'num_nextn_predict_layers'] # Current MTP only support linear tree + if not self.mtp_eagle_one_model: + logger.warning( + "2-model style MTP is deprecated. The mtp_eagle_one_model flag will do nothing " + "in release 1.3. After that, the flag will be removed entirely." + ) + @classmethod def from_dict(cls, data: dict): out = cls(**data)