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

Update import_fixes.py

Browse source
This commit is contained in:
Daniel Han 2025-12-12 05:36:40 -08:00
commit e17b62f76e
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/import_fixes.py
View file

@ -268,6 +268,7 @@ def check_fbgemm_gpu_version():
)
logger.info(f"Unsloth: fbgemm_gpu_genai=={fbgemm_gpu_version} detected.")
print([mod for mod in ["trl", "transformers", "peft"] if mod in sys.modules])
def patch_enable_input_require_grads():
"""
@ -331,6 +332,7 @@ def patch_enable_input_require_grads():
"Unsloth: Patched enable_input_require_grads for vision model compatibility"
)
print([mod for mod in ["trl", "transformers", "peft"] if mod in sys.modules])
def torchvision_compatibility_check():
if importlib.util.find_spec("torch") is None: