ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix

Browse source
This commit is contained in:
Daniel Han 2025-12-12 05:27:42 -08:00
commit 65aaa0b01c
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/__init__.py
View file

@ -124,7 +124,7 @@ from .import_fixes import (
patch_ipykernel_hf_xet,
patch_trackio,
patch_datasets,
patch_enable_input_require_grads,
# patch_enable_input_require_grads,
fix_openenv_no_vllm,
)
@ -135,7 +135,7 @@ ignore_logger_messages()
patch_ipykernel_hf_xet()
patch_trackio()
patch_datasets()
patch_enable_input_require_grads()
# patch_enable_input_require_grads()
fix_openenv_no_vllm()
del fix_xformers_performance_issue

4
unsloth/import_fixes.py
View file

@ -386,8 +386,8 @@ def fix_openenv_no_vllm():
text = f.read()
bad = (
"if is_vllm_available():\n"
"from vllm import SamplingParams\n"
"from vllm.sampling_params import GuidedDecodingParams\n"
" from vllm import SamplingParams\n"
" from vllm.sampling_params import GuidedDecodingParams\n"
)
if bad + "\n" + "\n" in text:
text = text.replace(