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

Update __init__.py

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

2
unsloth/__init__.py
View file

@ -16,6 +16,7 @@ import warnings, importlib, sys
from packaging.version import Version
import os, re, subprocess, inspect, functools
import numpy as np
print([mod for mod in ["trl", "transformers", "peft"] if mod in sys.modules])
# Log Unsloth is being used
os.environ["UNSLOTH_IS_PRESENT"] = "1"
@ -26,6 +27,7 @@ from .import_fixes import (
check_fbgemm_gpu_version,
torchvision_compatibility_check,
)
print([mod for mod in ["trl", "transformers", "peft"] if mod in sys.modules])
fix_message_factory_issue()
check_fbgemm_gpu_version()