diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 8b23e0c837..aa8e128ca7 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -139,6 +139,7 @@ warnings.filterwarnings(action = "ignore", category = UserWarning, module = " warnings.filterwarnings(action = "ignore", category = FutureWarning, module = "accelerate") warnings.filterwarnings(action = "ignore", category = RuntimeWarning, module = "multiprocessing") warnings.filterwarnings(action = "ignore", category = RuntimeWarning, module = "multiprocess") +warnings.filterwarnings(action = "ignore", category = UserWarning, module = "triton") # Stop "Special tokens have been added in the vocabulary, ..." import logging