fix

Author: suzukimain

src/diffusers/pipelines/pipeline_utils.py

@@ -1987,18 +1987,25 @@ def unfuse_qkv_projections(self, unet: bool = True, vae: bool = True):
                 self.vae.unfuse_qkv_projections()
                 self.fusing_vae = False
 
-    def safety_checker_Level(self, Level):
+    def safety_checker_level(self, level):
         """
         Adjust the filter intensity.
         
         Args:
-            Level (`int` or `float` or one of the following [`WEAK`], [`MEDIUM`], [`NOMAL`], [`STRONG`], [`MAX`])
+            level (`int` or `float` or one of the following [`WEAK`], [`MEDIUM`], [`NOMAL`], [`STRONG`], [`MAX`])
         """
         _safety_checker = getattr(self, "safety_checker", None)
         if _safety_checker is not None:
             if hasattr(_safety_checker, "update_safety_checker_Level"):
                 self.safety_checker.update_safety_checker_Level(Level)
             else:
-                logger.warning("`safety_checker_Level` is ignored because `update_safety_checker_Level` is not in `safety_checker`.")
+                logger.warning("`safety_checker_level` is ignored because `update_safety_checker_Level` is not in `safety_checker`.")
         else:
-            logger.warning("Since there is no `safety_checker`, `safety_checker_Level` is ignored.")
+            logger.warning("Since there is no `safety_checker`, `safety_checker_level` is ignored.")
+
+    def safety_level(self):
+        """
+        Return:
+            `int` ,`float` or None
+        """
+        return getattr(getattr(self,"safety_checker",None), "adjustment", None)

src/diffusers/pipelines/stable_diffusion/safety_checker.py

@@ -34,7 +34,6 @@ class StableDiffusionSafetyChecker(PreTrainedModel):
     main_input_name = "clip_input"
 
     _no_split_modules = ["CLIPEncoderLayer"]
-    
 
     def __init__(self, config: CLIPConfig):
         super().__init__(config)
@@ -63,7 +62,7 @@ def update_safety_checker_Level(self, Level):
             "MAX": 1.0,
             }
         if Level in Level_dict:
-            Level = Level_dict[Level]     
+            Level = Level_dict[Level]
         if isinstance(Level, (float, int)): 
             setattr(self,"adjustment",Level)
         else: