From 59a5ba9be0c7d2453aeacf61ac20bad99c04ec10 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Mon, 11 Nov 2024 15:36:14 -0800 Subject: [PATCH] [Minor] Remove unused imports (#2006) --- python/sglang/srt/models/gemma2_reward.py | 1 - python/sglang/srt/models/llama_reward.py | 2 -- 2 files changed, 3 deletions(-) diff --git a/python/sglang/srt/models/gemma2_reward.py b/python/sglang/srt/models/gemma2_reward.py index 9aab3ce18..d794d8477 100644 --- a/python/sglang/srt/models/gemma2_reward.py +++ b/python/sglang/srt/models/gemma2_reward.py @@ -18,7 +18,6 @@ from typing import Iterable, Optional, Tuple import torch from torch import nn from transformers import Gemma2Config -from vllm.model_executor.model_loader.weight_utils import default_weight_loader from sglang.srt.layers.pooler import EmbeddingPoolerOutput, Pooler, PoolingType from sglang.srt.layers.quantization.base_config import QuantizationConfig diff --git a/python/sglang/srt/models/llama_reward.py b/python/sglang/srt/models/llama_reward.py index e285ad692..f84a4b225 100644 --- a/python/sglang/srt/models/llama_reward.py +++ b/python/sglang/srt/models/llama_reward.py @@ -18,9 +18,7 @@ from typing import Iterable, Optional, Tuple import torch from torch import nn from transformers import LlamaConfig -from vllm.model_executor.model_loader.weight_utils import default_weight_loader -from sglang.srt.layers.logits_processor import LogitsProcessorOutput from sglang.srt.layers.pooler import EmbeddingPoolerOutput, Pooler, PoolingType from sglang.srt.layers.quantization.base_config import QuantizationConfig from sglang.srt.model_executor.forward_batch_info import ForwardBatch