From a81596dc0ca9fcd9f138ddd5abf3e0ae54db6adc Mon Sep 17 00:00:00 2001 From: Hrishikesh Date: Mon, 16 Dec 2024 17:53:56 -0800 Subject: [PATCH] disable pylint warning --- python/mlc_llm/model/nemotron/nemotron_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/mlc_llm/model/nemotron/nemotron_model.py b/python/mlc_llm/model/nemotron/nemotron_model.py index 4fe9363741..7d22a2f4b8 100644 --- a/python/mlc_llm/model/nemotron/nemotron_model.py +++ b/python/mlc_llm/model/nemotron/nemotron_model.py @@ -45,7 +45,7 @@ class NemotronConfig(ConfigBase): # pylint: disable=too-many-instance-attribute disaggregation: bool = False kwargs: Dict[str, Any] = dataclasses.field(default_factory=dict) - def __post_init__(self): + def __post_init__(self): # pylint: disable=too-many-branches if self.context_window_size == 0: self.context_window_size = self.max_position_embeddings if self.head_dim == 0: @@ -219,7 +219,7 @@ def forward(self, input_embed: Tensor, paged_kv_cache: PagedKVCache): return hidden_states -class NemotronForCausalLM(nn.Module): +class NemotronForCausalLM(nn.Module): # pylint: disable=too-many-instance-attributes def __init__(self, config: NemotronConfig): self.model = NemotronModel(config) self.tie_word_embeddings = config.tie_word_embeddings