Skip to content

Commit 0c0b4a2

Browse files
committed
delete useless code in sglang.py
1 parent 9219961 commit 0c0b4a2

File tree

2 files changed

+2
-7
lines changed

2 files changed

+2
-7
lines changed

xtuner/v1/data_proto/rl_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -429,7 +429,7 @@ class SampleParams(BaseModel):
429429
stops: Annotated[list[str], Parameter(help="List of stop sequences.")] = []
430430
stop_token_ids: Annotated[list[int], Parameter(help="List of stop token IDs.")] = []
431431
skip_special_tokens: Annotated[bool, Parameter(help="Whether to skip special tokens.")] = True
432-
sampling_seed: Annotated[int, Parameter(help="The seed for random number generator in sampling.")] = 0
432+
sampling_seed: Annotated[int | None, Parameter(help="The seed for random number generator in sampling.")] = None
433433

434434

435435
class RolloutExtraParams(TypedDict):

xtuner/v1/ray/rollout/sglang.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import requests
77
from urllib3.exceptions import NewConnectionError
88

9-
from transformers import AutoConfig, AutoTokenizer
9+
from transformers import AutoTokenizer
1010
from xtuner.v1.ray.config import RolloutConfig
1111
from xtuner.v1.utils import XTUNER_DETERMINISTIC
1212

@@ -31,11 +31,6 @@ def __init__(
3131
self.endpoints["generate"] = "generate"
3232
self.endpoints["v1/chat/completions"] = "v1/chat/completions"
3333
self.tokenizer = AutoTokenizer.from_pretrained(self.config.model_path, trust_remote_code=True)
34-
self.model_config = AutoConfig.from_pretrained(self.config.model_path, trust_remote_code=True)
35-
text_config = getattr(self.model_config, "text_config", self.model_config)
36-
self.model_type = getattr(text_config, "model_type", getattr(self.model_config, "model_type", None))
37-
self.routed_experts_num_hidden_layers = getattr(text_config, "num_hidden_layers", None)
38-
self.routed_experts_num_experts_per_tok = getattr(text_config, "num_experts_per_tok", None)
3934
self.api_keys = self.config.api_key
4035
self.model_name = self.config.model_name
4136
self.enable_return_routed_experts = self.config.enable_return_routed_experts

0 commit comments

Comments
 (0)