diff --git a/comfy/utils.py b/comfy/utils.py
index 0dccea7f0..84d1d0c9d 100644
--- a/comfy/utils.py
+++ b/comfy/utils.py
@@ -57,9 +57,9 @@ if hasattr(torch.serialization, "add_safe_globals"):  # TODO: this was added in
 
     ModelCheckpoint.__module__ = "pytorch_lightning.callbacks.model_checkpoint"
 
-    from numpy.core.multiarray import scalar
+    from numpy.core.multiarray import scalar  # pylint: disable=no-name-in-module
     from numpy import dtype
-    from numpy.dtypes import Float64DType
+    from numpy.dtypes import Float64DType  # pylint: disable=no-name-in-module
     from _codecs import encode
 
     torch.serialization.add_safe_globals([ModelCheckpoint, scalar, dtype, Float64DType, encode])
diff --git a/requirements.txt b/requirements.txt
index bc32316d4..9c38346c3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -69,4 +69,5 @@ skia-python
 pebble>=5.0.7
 openai
 anthropic
-humanize
\ No newline at end of file
+humanize
+lightning
\ No newline at end of file