cleanup deps 3/n (#4541)

This commit is contained in:
Yineng Zhang
2025-03-18 00:11:36 -07:00
committed by GitHub
parent 2d0045125f
commit c16b33ccac
4 changed files with 16 additions and 12 deletions
+2 -1
View File
@@ -18,7 +18,6 @@ from typing import Iterable, Optional, Tuple
import torch
from torch import nn
from transformers import PretrainedConfig
from vllm import _custom_ops as ops
from sglang.srt.layers.layernorm import RMSNorm
from sglang.srt.layers.linear import ReplicatedLinear
@@ -48,6 +47,8 @@ _is_cuda = is_cuda()
if _is_cuda:
from sgl_kernel import awq_dequantize
else:
from vllm import _custom_ops as ops
class DeepseekModelNextN(nn.Module):