minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Replace single-line defensive getattrs with direct access (#25435)

Browse Source
This commit is contained in:
fzyzcjy
2026-05-16 09:18:43 +08:00
committed by GitHub
parent 27c3190215
commit 6a1b05d706
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/managers/scheduler_output_processor_mixin.py
+1 -1
View File
@@ -75,7 +75,7 @@ class SchedulerOutputProcessorMixin:
"host": req.cached_tokens_host, "host": req.cached_tokens_host,
} }
# Only include storage fields if L3 storage is enabled # Only include storage fields if L3 storage is enabled
if getattr(self, "enable_hicache_storage", False): if self.enable_hicache_storage:
details["storage"] = req.cached_tokens_storage details["storage"] = req.cached_tokens_storage
details["storage_backend"] = self._get_storage_backend_type() details["storage_backend"] = self._get_storage_backend_type()
return details return details
python/sglang/srt/managers/scheduler_pp_mixin.py
+1 -1
View File
@@ -717,7 +717,7 @@ class SchedulerPPMixin:
): ):
return None return None
max_chunk_size = getattr(self, "max_prefill_tokens", None) max_chunk_size = self.max_prefill_tokens
predicted_size = self.length_predictor.predict_next_chunk_size( predicted_size = self.length_predictor.predict_next_chunk_size(
history_len=history_len, history_len=history_len,
base_chunk_size=self.chunked_prefill_size, base_chunk_size=self.chunked_prefill_size,
python/sglang/srt/managers/tokenizer_manager_score_mixin.py
+3 -3
View File
@@ -261,7 +261,7 @@ class TokenizerManagerScoreMixin:
has_phs = False has_phs = False
prompt_tokens = 0 prompt_tokens = 0
is_generation = getattr(self, "is_generation", True) is_generation = self.is_generation
if is_generation: if is_generation:
for result in results: for result in results:
# For single-item scoring, logprobs are in output_token_ids_logprobs # For single-item scoring, logprobs are in output_token_ids_logprobs
@@ -563,7 +563,7 @@ class TokenizerManagerScoreMixin:
return_pooled_hidden_states=True and the model supports it; return_pooled_hidden_states=True and the model supports it;
None otherwise. None otherwise.
""" """
is_generation = getattr(self, "is_generation", True) is_generation = self.is_generation
if is_generation and label_token_ids is None: if is_generation and label_token_ids is None:
raise ValueError( raise ValueError(
@@ -676,7 +676,7 @@ class TokenizerManagerScoreMixin:
"It requires a model with a task-specific head " "It requires a model with a task-specific head "
"(e.g. SequenceClassification or RewardModel)." "(e.g. SequenceClassification or RewardModel)."
) )
model_config = getattr(self, "model_config", None) model_config = self.model_config
if model_config is not None: if model_config is not None:
archs = getattr(model_config.hf_config, "architectures", []) or [] archs = getattr(model_config.hf_config, "architectures", []) or []
if is_cross_encoding_pooler_model(archs): if is_cross_encoding_pooler_model(archs):