mirror of
https://github.com/vladmandic/sdnext.git
synced 2026-01-29 05:02:09 +03:00
39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
from __future__ import annotations
|
|
import re
|
|
import logging
|
|
import warnings
|
|
import urllib3
|
|
from modules import timer, errors
|
|
|
|
initialized = False
|
|
logging.getLogger("DeepSpeed").disabled = True
|
|
import torch # pylint: disable=C0411
|
|
errors.log.debug(f'Loaded Torch=={torch.__version__}')
|
|
try:
|
|
import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import
|
|
errors.log.debug(f'Loaded IPEX=={ipex.__version__}')
|
|
except Exception:
|
|
pass
|
|
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
|
import torchvision # pylint: disable=W0611,C0411
|
|
import pytorch_lightning # pytorch_lightning should be imported after torch, but it re-enables warnings on import so import once to disable them # pylint: disable=W0611,C0411
|
|
if ".dev" in torch.__version__ or "+git" in torch.__version__:
|
|
torch.__long_version__ = torch.__version__
|
|
torch.__version__ = re.search(r'[\d.]+[\d]', torch.__version__).group(0)
|
|
logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
|
|
logging.getLogger("pytorch_lightning").disabled = True
|
|
warnings.filterwarnings(action="ignore", category=DeprecationWarning)
|
|
warnings.filterwarnings(action="ignore", category=FutureWarning)
|
|
warnings.filterwarnings(action="ignore", category=UserWarning, module="torchvision")
|
|
timer.startup.record("torch")
|
|
|
|
from fastapi import FastAPI # pylint: disable=W0611,C0411
|
|
import gradio # pylint: disable=W0611,C0411
|
|
errors.log.debug(f'Loaded Gradio=={gradio.__version__}')
|
|
timer.startup.record("gradio")
|
|
errors.install([gradio])
|
|
|
|
import diffusers # pylint: disable=W0611,C0411
|
|
errors.log.debug(f'Loaded Diffusers=={diffusers.__version__}')
|
|
timer.startup.record("diffusers")
|