From be77c66a84022605f56c11e1f592f855e299f7fd Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 15 Feb 2026 23:02:12 -0800 Subject: [PATCH] Add reinstall command to broken vLLM warning (#4070) * Add vLLM reinstall command to broken-extension warning * Apply suggestion from @danielhanchen --------- Co-authored-by: Daniel Hanchen --- unsloth/import_fixes.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/unsloth/import_fixes.py b/unsloth/import_fixes.py index 0581cf7bdf..6e5826f6b8 100644 --- a/unsloth/import_fixes.py +++ b/unsloth/import_fixes.py @@ -1526,7 +1526,9 @@ def disable_broken_vllm(error = None): _install_vllm_blocker() logger.warning( "Unsloth: Detected broken vLLM binary extension; " - "disabling vLLM imports and continuing import." + "disabling vLLM imports and continuing import.\n" + "Please reinstall via `uv pip install unsloth vllm torchvision torchaudio " + "--torch-backend=auto`." ) return True