Commit d9b9bf78 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub

Merge pull request #8508 from vladmandic/lightning

allow usage of latest pytorch_lighning
parents 009bc9f5 f2ed6295
...@@ -9,7 +9,7 @@ Pillow==9.4.0 ...@@ -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
......
...@@ -4,6 +4,7 @@ import time ...@@ -4,6 +4,7 @@ import time
import importlib import importlib
import signal import signal
import re import re
import warnings
from fastapi import FastAPI from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from fastapi.middleware.gzip import GZipMiddleware from fastapi.middleware.gzip import GZipMiddleware
...@@ -17,6 +18,8 @@ from modules import paths, timer, import_hook, errors ...@@ -17,6 +18,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, module="pytorch_lightning")
startup_timer.record("import torch") startup_timer.record("import torch")
import gradio import gradio
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment