Merge pull request #8508 from vladmandic/lightning
allow usage of latest pytorch_lighning
This commit is contained in:
commit
d9b9bf78b3
2 changed files with 4 additions and 1 deletions
|
@ -9,7 +9,7 @@ Pillow==9.4.0
|
|||
realesrgan==0.3.0
|
||||
torch
|
||||
omegaconf==2.2.3
|
||||
pytorch_lightning==1.7.6
|
||||
pytorch_lightning==1.9.4
|
||||
scikit-image==0.19.2
|
||||
fonts
|
||||
font-roboto
|
||||
|
|
3
webui.py
3
webui.py
|
@ -4,6 +4,7 @@ import time
|
|||
import importlib
|
||||
import signal
|
||||
import re
|
||||
import warnings
|
||||
from fastapi import FastAPI
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from fastapi.middleware.gzip import GZipMiddleware
|
||||
|
@ -17,6 +18,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, module="pytorch_lightning")
|
||||
startup_timer.record("import torch")
|
||||
|
||||
import gradio
|
||||
|
|
Loading…
Reference in a new issue