make it module specific

3 years ago · f2ed6295b9
parent fc4d593b4e
commit f2ed6295b9
1 changed files with 1 additions and 1 deletions
--- a/webui.py
+++ b/webui.py
@ -19,7 +19,7 @@ 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)
+warnings.filterwarnings(action="ignore", category=DeprecationWarning, module="pytorch_lightning")
 startup_timer.record("import torch")

 import gradio