Improve code style of sampler (#1168)

2024-08-21 16:48:24 -07:00
parent ac1b74fa85
commit 83e23c69b3
10 changed files with 268 additions and 194 deletions
--- a/python/sglang/srt/managers/io_struct.py
+++ b/python/sglang/srt/managers/io_struct.py
@@ -23,7 +23,7 @@ from dataclasses import dataclass
 from typing import Dict, List, Optional, Union

 from sglang.srt.managers.schedule_batch import BaseFinishReason
-from sglang.srt.sampling_params import SamplingParams
+from sglang.srt.sampling.sampling_params import SamplingParams


@dataclass