[lint] improve ruff check (#11922)

Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
This commit is contained in:
Liangsheng Yin
2025-10-22 11:32:50 +08:00
committed by GitHub
co-authored by Xiaoyu Zhang
parent 590bc4b7a7
commit 9d61205dac
19 changed files with 73 additions and 31 deletions
+2 -1
View File
@@ -25,6 +25,7 @@ from typing import Any, Dict, Iterable, Optional, Tuple, Union
import torch
import torch.nn.functional as F
import tqdm
from torch import nn
from transformers import PretrainedConfig
@@ -3499,7 +3500,7 @@ class DeepseekV2ForCausalLM(nn.Module):
# temporarily only support DeepSeek V3/R1
weight_block_size = [128, 128]
for layer_id in trange(
for layer_id in tqdm.trange(
self.config.num_hidden_layers + int(is_nextn),
desc="quant attn to fp8 ue8m0",
):