all usage of newer pytorch_lighning

This commit is contained in:
Vladimir Mandic
2023-03-11 11:13:21 -05:00
committed by GitHub
parent 94ffa9fc53
commit fb088bfb64
2 changed files with 3 additions and 1 deletions

View File

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