Unverified Commit 2d867b55 authored by MoeedDar's avatar MoeedDar Committed by GitHub
Browse files

fixed tensor parallel is not defined (#564)

parent d7a1c6d6
import vllm.model_executor.parallel_utils.parallel_state
import vllm.model_executor.parallel_utils.tensor_parallel
# Alias parallel_state as mpu, its legacy name
mpu = parallel_state
__all__ = [
"parallel_state",
"tensor_parallel",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment