Code formatting

Author: suzukimain

src/diffusers/pipelines/pipeline_utils.py

@@ -1939,23 +1939,33 @@ def unfuse_qkv_projections(self, unet: bool = True, vae: bool = True):
 
     def safety_checker_level(self, level):
         """
-        Adjust the filter intensity.
-        
+        Adjust the safety checker level.
+
         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`]):
+                The level of safety checker adjustment, either as an integer, a float, or one of the predefined levels.
+                Negative values decrease the filtering strength, while positive values increase it.
         """
+        # Retrieve the safety_checker attribute from the instance
         _safety_checker = getattr(self, "safety_checker", None)
+
+        # Check if the safety_checker exists
         if _safety_checker is not None:
+            # Check if the safety_checker has the update_safety_checker_Level method
             if hasattr(_safety_checker, "update_safety_checker_Level"):
+                # Update the safety checker level using the provided method
                 self.safety_checker.update_safety_checker_Level(level)
             else:
+                # Log a warning if the method is not found in safety_checker
                 logger.warning("`safety_checker_level` is ignored because `update_safety_checker_Level` is not in `safety_checker`.")
         else:
+            # Log a warning if safety_checker is not present
             logger.warning("Since there is no `safety_checker`, `safety_checker_level` is ignored.")
 
+    @property
     def filter_level(self):
         """
         Return:
             `int` ,`float` or None
         """
-        return getattr(getattr(self,"safety_checker",None), "adjustment", None)
+        return getattr(getattr(self, "safety_checker", None), "adjustment", None)

src/diffusers/pipelines/stable_diffusion/safety_checker.py

@@ -51,8 +51,12 @@ def __init__(self, config: CLIPConfig):
 
     def update_safety_checker_Level(self, Level):
         """
-        Args:
-            Level (`int` or `float` or one of the following [`WEAK`], [`MEDIUM`], [`NOMAL`], [`STRONG`], [`MAX`])
+        Adjust the safety checker level.
+
+        Parameters:
+            Level (`int` or `float` or one of the following [`WEAK`], [`MEDIUM`], [`NOMAL`], [`STRONG`], [`MAX`]):
+                The level of safety checker adjustment, either as an integer, a float, or one of the predefined levels.
+                Negative values decrease the filtering strength, while positive values increase it.
         """
         Level_dict = {
             "WEAK": -0.10,
@@ -61,16 +65,22 @@ def update_safety_checker_Level(self, Level):
             "STRONG": 0.01,
             "MAX": 0.10,
             }
+
+        # If the provided Level is a predefined string, convert it to the corresponding float value
         if Level in Level_dict:
             Level = Level_dict[Level]
+
+        # Check if the Level is a float or an integer
         if isinstance(Level, (float, int)): 
-            setattr(self,"adjustment",Level)
+            setattr(self, "adjustment", Level)  # Set the adjustment attribute to the Level value
         else:
+            # Raise an error if Level is not a valid type or predefined string
             raise ValueError("`int` or `float` or one of the following ['WEAK'], ['MEDIUM'], ['NOMAL'], ['STRONG'], ['MAX']")
-        
-        if self.adjustment<0:
+
+        # Log a warning if the adjustment level is weakened (negative value)
+        if self.adjustment < 0:
             logger.warning(
-                f"You have weakened the filtering strength of safety checker. Ensure"
+                "You have weakened the filtering strength of safety checker. Ensure"
                 " that you abide to the conditions of the Stable Diffusion license and do not expose unfiltered"
                 " results in services or applications open to the public. Both the diffusers team and Hugging Face"
                 " strongly recommend to keep the safety filter enabled in all public facing circumstances, disabling"