safety checker

Author: a-r-r-o-w

src/diffusers/pipelines/cosmos/pipeline_cosmos2_text2image.py

@@ -171,8 +171,8 @@ def __init__(
     ):
         super().__init__()
 
-        # if safety_checker is None:
-        #     safety_checker = CosmosSafetyChecker()
+        if safety_checker is None:
+            safety_checker = CosmosSafetyChecker()
 
         self.register_modules(
             vae=vae,
@@ -474,12 +474,12 @@ def __call__(
                 indicating whether the corresponding generated image contains "not-safe-for-work" (nsfw) content.
         """
 
-        # if self.safety_checker is None:
-        #     raise ValueError(
-        #         f"You have disabled the safety checker for {self.__class__}. This is in violation of the "
-        #         "[NVIDIA Open Model License Agreement](https://www.nvidia.com/en-us/agreements/enterprise-software/nvidia-open-model-license). "
-        #         f"Please ensure that you are compliant with the license agreement."
-        #     )
+        if self.safety_checker is None:
+            raise ValueError(
+                f"You have disabled the safety checker for {self.__class__}. This is in violation of the "
+                "[NVIDIA Open Model License Agreement](https://www.nvidia.com/en-us/agreements/enterprise-software/nvidia-open-model-license). "
+                f"Please ensure that you are compliant with the license agreement."
+            )
 
         if isinstance(callback_on_step_end, (PipelineCallback, MultiPipelineCallbacks)):
             callback_on_step_end_tensor_inputs = callback_on_step_end.tensor_inputs

src/diffusers/pipelines/cosmos/pipeline_cosmos2_video2world.py

@@ -190,8 +190,8 @@ def __init__(
     ):
         super().__init__()
 
-        # if safety_checker is None:
-        #     safety_checker = CosmosSafetyChecker()
+        if safety_checker is None:
+            safety_checker = CosmosSafetyChecker()
 
         self.register_modules(
             vae=vae,
@@ -561,12 +561,12 @@ def __call__(
                 indicating whether the corresponding generated image contains "not-safe-for-work" (nsfw) content.
         """
 
-        # if self.safety_checker is None:
-        #     raise ValueError(
-        #         f"You have disabled the safety checker for {self.__class__}. This is in violation of the "
-        #         "[NVIDIA Open Model License Agreement](https://www.nvidia.com/en-us/agreements/enterprise-software/nvidia-open-model-license). "
-        #         f"Please ensure that you are compliant with the license agreement."
-        #     )
+        if self.safety_checker is None:
+            raise ValueError(
+                f"You have disabled the safety checker for {self.__class__}. This is in violation of the "
+                "[NVIDIA Open Model License Agreement](https://www.nvidia.com/en-us/agreements/enterprise-software/nvidia-open-model-license). "
+                f"Please ensure that you are compliant with the license agreement."
+            )
 
         if isinstance(callback_on_step_end, (PipelineCallback, MultiPipelineCallbacks)):
             callback_on_step_end_tensor_inputs = callback_on_step_end.tensor_inputs