fixed tensor parallel is not defined (#564)

This commit is contained in:
MoeedDar 2023-07-25 22:16:51 +01:00 committed by GitHub
parent d7a1c6d614
commit 2d867b55fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,9 +1,6 @@
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",