Add warning when using old pytorch. (Comfy-Org#9347)

comfyanonymous · web-flow · commit e08ecfbd8a9d · 2025-08-15T00:22:26.000-04:00
diff --git a/comfy/rmsnorm.py b/comfy/rmsnorm.py
@@ -1,6 +1,7 @@
 import torch
 import comfy.model_management
 import numbers
+import logging
 
 RMSNorm = None
 
@@ -9,6 +10,7 @@
     RMSNorm = torch.nn.RMSNorm
 except:
     rms_norm_torch = None
+    logging.warning("Please update pytorch to use native RMSNorm")
 
 
 def rms_norm(x, weight=None, eps=1e-6):
