Fix import error msg for megablocks (#2038)

vllm-project · Dec 11, 2023 · 6120e5a · 6120e5a
1 parent 2eaa81b
commit 6120e5a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vllm/model_executor/models/mixtral.py b/vllm/model_executor/models/mixtral.py
@@ -34,10 +34,10 @@
 try:
     import megablocks.ops as ops
 except ImportError:
-    print("MegaBlocks not found, please see "
-          "https://github.com/stanford-futuredata/megablocks/. "
-          "Note that MegaBlocks depends on mosaicml-turbo, which only "
-          "supports python 3.10.")
+    print(
+        "MegaBlocks not found. Please install it by `pip install megablocks`. "
+        "Note that MegaBlocks depends on mosaicml-turbo, which only supports "
+        "Python 3.10 for now.")
 try:
     import stk
 except ImportError: