Skip to content

Commit 6120e5a

Browse files
authored
Fix import error msg for megablocks (#2038)
1 parent 2eaa81b commit 6120e5a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

vllm/model_executor/models/mixtral.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,10 +34,10 @@
3434
try:
3535
import megablocks.ops as ops
3636
except ImportError:
37-
print("MegaBlocks not found, please see "
38-
"https://github.com/stanford-futuredata/megablocks/. "
39-
"Note that MegaBlocks depends on mosaicml-turbo, which only "
40-
"supports python 3.10.")
37+
print(
38+
"MegaBlocks not found. Please install it by `pip install megablocks`. "
39+
"Note that MegaBlocks depends on mosaicml-turbo, which only supports "
40+
"Python 3.10 for now.")
4141
try:
4242
import stk
4343
except ImportError:

0 commit comments

Comments
 (0)