From f34d08ec6ab1ce2b6678bf683bc2e90412697a5e Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 17 Dec 2025 03:25:40 -0800
Subject: [PATCH] Diffusers warnings

---
 unsloth/__init__.py     | 3 +++
 unsloth/import_fixes.py | 5 +++++
 2 files changed, 8 insertions(+)

diff --git a/unsloth/__init__.py b/unsloth/__init__.py
index bf3de82dc0..d10a0f8030 100644
--- a/unsloth/__init__.py
+++ b/unsloth/__init__.py
@@ -29,14 +29,17 @@ from .import_fixes import (
     fix_message_factory_issue,
     check_fbgemm_gpu_version,
     torchvision_compatibility_check,
+    fix_diffusers_warnings,
 )
 
 fix_message_factory_issue()
 check_fbgemm_gpu_version()
 torchvision_compatibility_check()
+fix_diffusers_warnings()
 del fix_message_factory_issue
 del check_fbgemm_gpu_version
 del torchvision_compatibility_check
+del fix_diffusers_warnings
 
 # This check is critical because Unsloth optimizes these libraries by modifying
 # their code at import time. If they're imported first, the original (slower,
diff --git a/unsloth/import_fixes.py b/unsloth/import_fixes.py
index 308bd92db7..f3aae7f523 100644
--- a/unsloth/import_fixes.py
+++ b/unsloth/import_fixes.py
@@ -536,3 +536,8 @@ def fix_executorch():
                 logger.info("Unsloth: Patching Executorch to fix get_mapped_key")
     except Exception as e:
         logger.info(f"Unsloth: Failed Executorch with error = {str(e)}")
+
+
+def fix_diffusers_warnings():
+    # Silence Flax classes are deprecated and will be removed in Diffusers v1.0.0.
+    os.environ["DIFFUSERS_VERBOSITY"] = "error"