cleanup deps 3/n (#4541)
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user