Skip to content

Commit

Permalink
misc: rm unused model_loader (#1110)
Browse files Browse the repository at this point in the history
  • Loading branch information
zhyncs authored Aug 15, 2024
1 parent 14cb544 commit 9195d13
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 576 deletions.
10 changes: 1 addition & 9 deletions python/sglang/srt/model_executor/model_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@
init_distributed_environment,
initialize_model_parallel,
)
from vllm.model_executor.model_loader import get_model
from vllm.model_executor.models import ModelRegistry

from sglang.global_config import global_config
Expand Down Expand Up @@ -168,15 +169,6 @@ def load_model(self):
if self.model_config.model_overide_args is not None:
vllm_model_config.hf_config.update(self.model_config.model_overide_args)

if (
self.server_args.efficient_weight_load
and "llama" in self.server_args.model_path.lower()
and self.server_args.quantization == "fp8"
):
from sglang.srt.model_loader.model_loader import get_model
else:
from vllm.model_executor.model_loader import get_model

self.model = get_model(
model_config=vllm_model_config,
device_config=device_config,
Expand Down
292 changes: 0 additions & 292 deletions python/sglang/srt/model_loader/model_loader.py

This file was deleted.

Loading

0 comments on commit 9195d13

Please sign in to comment.