diff options
-rw-r--r-- | requirements_versions.txt | 2 | ||||
-rw-r--r-- | webui.py | 2 |
2 files changed, 3 insertions, 1 deletions
diff --git a/requirements_versions.txt b/requirements_versions.txt index 41e0ccc5..b98081bf 100644 --- a/requirements_versions.txt +++ b/requirements_versions.txt @@ -9,7 +9,7 @@ Pillow==9.4.0 realesrgan==0.3.0
torch
omegaconf==2.2.3
-pytorch_lightning==1.7.6
+pytorch_lightning==1.9.4
scikit-image==0.19.2
fonts
font-roboto
@@ -17,6 +17,8 @@ from modules import paths, timer, import_hook, errors startup_timer = timer.Timer()
import torch
+import pytorch_lightning # pytorch_lightning should be imported after torch, but it re-enables warnings on import so import once to disable them
+warnings.filterwarnings(action="ignore", category=DeprecationWarning)
startup_timer.record("import torch")
import gradio
|