Merge pull request #9191 from vladmandic/torch

update torch base environment

1 files changed, 2 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index 0e2a3df0..1f97b475 100644
--- a/webui.py
+++ b/webui.py
@@ -20,6 +20,8 @@ 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, module="pytorch_lightning")

+warnings.filterwarnings(action="ignore", category=UserWarning, module="torchvision")

+

 startup_timer.record("import torch")

 

 import gradio