make it module specific

This commit is contained in:
Vladimir Mandic
2023-03-14 07:46:09 -04:00
committed by GitHub
parent fc4d593b4e
commit f2ed6295b9

View File

@@ -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