wrap cpu warning in reference_compile

Author: warner-benjamin

src/transformers/models/modernbert/modeling_modernbert.py

@@ -671,10 +671,11 @@ def _maybe_set_compile(self):
             self.config.reference_compile = False
 
         if self.device.type == "cpu":
-            logger.warning_once(
-                "Compiling the model with `torch.compile` and using a `torch.cpu` device is not supported. "
-                "Falling back to non-compiled mode."
-            )
+            if self.config.reference_compile:
+                logger.warning_once(
+                    "Compiling the model with `torch.compile` and using a `torch.cpu` device is not supported. "
+                    "Falling back to non-compiled mode."
+                )
             self.config.reference_compile = False
 
         if self.config.reference_compile is None:

src/transformers/models/modernbert/modular_modernbert.py

@@ -901,10 +901,11 @@ def _maybe_set_compile(self):
             self.config.reference_compile = False
 
         if self.device.type == "cpu":
-            logger.warning_once(
-                "Compiling the model with `torch.compile` and using a `torch.cpu` device is not supported. "
-                "Falling back to non-compiled mode."
-            )
+            if self.config.reference_compile:
+                logger.warning_once(
+                    "Compiling the model with `torch.compile` and using a `torch.cpu` device is not supported. "
+                    "Falling back to non-compiled mode."
+                )
             self.config.reference_compile = False
 
         if self.config.reference_compile is None: