fixed tensor parallel is not defined (#564)
This commit is contained in:
parent
d7a1c6d614
commit
2d867b55fa
@ -1,9 +1,6 @@
|
|||||||
import vllm.model_executor.parallel_utils.parallel_state
|
import vllm.model_executor.parallel_utils.parallel_state
|
||||||
import vllm.model_executor.parallel_utils.tensor_parallel
|
import vllm.model_executor.parallel_utils.tensor_parallel
|
||||||
|
|
||||||
# Alias parallel_state as mpu, its legacy name
|
|
||||||
mpu = parallel_state
|
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"parallel_state",
|
"parallel_state",
|
||||||
"tensor_parallel",
|
"tensor_parallel",
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user