Skip to content

GitLab

Unverified Commit 6120e5aa authored by Woosuk Kwon's avatar Woosuk Kwon Committed by GitHub
Browse files

Fix import error msg for megablocks (#2038)

parent 2eaa81b2
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
vllm/model_executor/models/mixtral.py
View file @ 6120e5aa
...@@ -34,10 +34,10 @@ from transformers import MistralConfig ...@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment