Sort imports

vllm/model_executor/layers/linear.py

@@ -5,6 +5,7 @@
 import torch.nn.functional as F
 from torch.nn.parameter import Parameter
 
+from vllm import _custom_C
 from vllm.logger import init_logger
 from vllm.model_executor.parallel_utils.communication_op import (
     tensor_model_parallel_all_gather, tensor_model_parallel_all_reduce)
@@ -14,7 +15,6 @@
     divide, split_tensor_along_last_dim)
 from vllm.model_executor.utils import set_weight_attrs
 from vllm.utils import is_hip
-from vllm import _custom_C
 
 logger = init_logger(__name__)
 

vllm/model_executor/models/llama.py

@@ -27,6 +27,7 @@
 from torch import nn
 from transformers import LlamaConfig
 
+from vllm import _custom_C
 from vllm.attention import Attention, AttentionMetadata
 from vllm.config import LoRAConfig
 from vllm.model_executor.layers.activation import SiluAndMul
@@ -46,7 +47,6 @@
 from vllm.model_executor.weight_utils import (default_weight_loader,
                                               hf_model_weights_iterator)
 from vllm.sequence import SamplerOutput
-from vllm import _custom_C
 
 
 class LlamaMLP(nn.Module):