[Feature] Make vLLM optional in model code

[ByronHsu]

UPDATE(11/23/2024)

Currently, @james-p-xu is removing rope, @yizhang2077 is removing distributed, @HandH1998 is removing weight loader. Optimistically, we can remove these dependencies by the end of the month and make quant optional (try import). cc @merrymercy @Ying1123

Motivation

This is a tracker of removing vLLM dependencies in general model code (not considering quantization). This is our current import from vLLM, and we want to remove all them.

from vllm.config import CacheConfig
from vllm.distributed import get_tensor_model_parallel_world_size
from vllm.model_executor.layers.rotary_embedding import get_rope
from vllm.model_executor.layers.vocab_parallel_embedding import (
   ParallelLMHead,
   VocabParallelEmbedding,
)

Tracker

• [x] Remove CacheConfig: https://github.com/sgl-project/sglang/pull/1658
• [ ] Remove RoPE: https://github.com/flashinfer-ai/flashinfer/issues/530
• [ ] Remove get_tensor_model_parallel_world_size
• [x] Remove ParallelLMHead: https://github.com/sgl-project/sglang/pull/1856
• [x] Remove VocabParallelEmbedding: https://github.com/sgl-project/sglang/pull/1856

[vkc1vk]

Just curious, are the following imports in model_runner.py also being considered for removal, in later stages

from vllm.config import DeviceConfig, LoadConfig
from vllm.config import ModelConfig as VllmModelConfig
from vllm.distributed import (
    get_tp_group,
    init_distributed_environment,
    initialize_model_parallel,
    set_custom_all_reduce,
)
from vllm.distributed.parallel_state import in_the_same_node_as
from vllm.model_executor.model_loader import get_model
from vllm.model_executor.models import ModelRegistry