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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions

View File

@ -9,7 +9,7 @@ Pillow==9.4.0
realesrgan==0.3.0 realesrgan==0.3.0
torch torch
omegaconf==2.2.3 omegaconf==2.2.3
pytorch_lightning==1.7.6 pytorch_lightning==1.9.4
scikit-image==0.19.2 scikit-image==0.19.2
fonts fonts
font-roboto font-roboto

View File

@ -17,6 +17,8 @@ from modules import paths, timer, import_hook, errors
startup_timer = timer.Timer() startup_timer = timer.Timer()
import torch 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") startup_timer.record("import torch")
import gradio import gradio