diff --git a/src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py b/src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py
index 581711205814..d23c4b42306d 100644
--- a/src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py
+++ b/src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py
@@ -664,7 +664,7 @@ def __call__(
 
         device = self._execution_device
 
-        if self.safety_checker is not None:
+        if isinstance(self.safety_checker, CosmosSafetyChecker):
             self.safety_checker.to(device)
             if prompt is not None:
                 prompt_list = [prompt] if isinstance(prompt, str) else prompt