squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix import error msg for megablocks (#2038)

Browse Source
This commit is contained in:
Woosuk Kwon 2023-12-11 11:40:56 -08:00 committed by GitHub
parent 2eaa81b236
commit 6120e5aaea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
vllm/model_executor/models/mixtral.py
View File

@ -34,10 +34,10 @@ from transformers import MistralConfig
try: try:
import megablocks.ops as ops import megablocks.ops as ops
except ImportError: except ImportError:
print("MegaBlocks not found, please see " print(
"https://github.com/stanford-futuredata/megablocks/. " "MegaBlocks not found. Please install it by `pip install megablocks`. "
"Note that MegaBlocks depends on mosaicml-turbo, which only " "Note that MegaBlocks depends on mosaicml-turbo, which only supports "
"supports python 3.10.") "Python 3.10 for now.")
try: try:
import stk import stk
except ImportError: except ImportError: