diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 990e4b915f..cc5c1d0474 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -301,8 +301,9 @@ elif DEVICE_TYPE == "xpu": # TODO: check triton for intel installed properly. pass -from .models import * -from .models import __version__ +with _suppress_hip_libdrm_ids_noise(): + from .models import * + from .models import __version__ from .save import * from .chat_templates import * from .tokenizer_utils import *