update torch

This commit is contained in:
Vladimir Mandic
2023-03-30 10:57:54 -04:00
parent 22bcc7be42
commit d5063e07e8
7 changed files with 16 additions and 12 deletions

View File

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