From 6120e5aaeaa336a5603ff0a1e64c189624098e9a Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Mon, 11 Dec 2023 11:40:56 -0800 Subject: [PATCH] Fix import error msg for megablocks (#2038) --- vllm/model_executor/models/mixtral.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/vllm/model_executor/models/mixtral.py b/vllm/model_executor/models/mixtral.py index f1482154..25d031db 100644 --- a/vllm/model_executor/models/mixtral.py +++ b/vllm/model_executor/models/mixtral.py @@ -34,10 +34,10 @@ from transformers import MistralConfig 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: