all usage of newer pytorch_lighning

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

diff --git a/webui.py b/webui.py
index 32561877..9baa7da5 100644
--- a/webui.py
+++ b/webui.py
@@ -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