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

Wrap models import with ROCm amdgpu ids fd2 filter (#4057)

Browse source 
Co-authored-by: Daniel Hanchen <danielhanchen@users.noreply.github.com>
This commit is contained in:
Daniel Han 2026-02-14 04:13:25 -08:00 • committed by GitHub
commit 842099f2b0
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/__init__.py
View file

@ -305,8 +305,10 @@ elif DEVICE_TYPE == "xpu":
# TODO: check triton for intel installed properly.
pass
from .models import *
from .models import __version__
# Filter native fd=2 amdgpu.ids noise during model import startup.
with _filter_rocm_amdgpu_ids_fd2_noise():
from .models import *
from .models import __version__
from .save import *
from .chat_templates import *
from .tokenizer_utils import *