mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
remove circular imports from installer
This commit is contained in:
+8
-16
@@ -55,7 +55,6 @@ def setup_logging(clean=False):
|
||||
# check if package is installed
|
||||
def installed(package, friendly: str = None):
|
||||
import pkg_resources
|
||||
from modules import shared
|
||||
ok = True
|
||||
try:
|
||||
if friendly:
|
||||
@@ -76,7 +75,7 @@ def installed(package, friendly: str = None):
|
||||
ok = ok and spec is not None
|
||||
if ok:
|
||||
version = pkg_resources.get_distribution(p[0]).version
|
||||
if shared.cmd_opts.use_ipex and p[0] == "pytorch_lightning":
|
||||
if args.use_ipex and p[0] == "pytorch_lightning":
|
||||
p[1] = "1.8.6"
|
||||
log.debug(f"Package version found: {p[0]} {version}")
|
||||
if len(p) > 1:
|
||||
@@ -93,8 +92,7 @@ def installed(package, friendly: str = None):
|
||||
|
||||
# install package using pip if not already installed
|
||||
def install(package, friendly: str = None, ignore: bool = False):
|
||||
from modules import shared
|
||||
if shared.cmd_opts.use_ipex and package == "pytorch_lightning==1.9.4":
|
||||
if args.use_ipex and package == "pytorch_lightning==1.9.4":
|
||||
package = "pytorch_lightning==1.8.6"
|
||||
def pip(arg: str):
|
||||
arg = arg.replace('>=', '==')
|
||||
@@ -192,7 +190,6 @@ def check_python():
|
||||
|
||||
# check torch version
|
||||
def check_torch():
|
||||
from modules import shared
|
||||
if shutil.which('nvidia-smi') is not None or os.path.exists(os.path.join(os.environ.get('SystemRoot') or r'C:\Windows', 'System32', 'nvidia-smi.exe')):
|
||||
log.info('nVidia toolkit detected')
|
||||
torch_command = os.environ.get('TORCH_COMMAND', 'torch torchaudio torchvision --index-url https://download.pytorch.org/whl/cu118')
|
||||
@@ -202,8 +199,7 @@ def check_torch():
|
||||
os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', '10.3.0')
|
||||
torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2')
|
||||
xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')
|
||||
elif shutil.which('sycl-ls') is not None or os.path.exists('/opt/intel/oneapi'):
|
||||
shared.cmd_opts.use_ipex = True
|
||||
elif shutil.which('sycl-ls') is not None or os.path.exists('/opt/intel/oneapi') or args.use_ipex:
|
||||
log.info('Intel toolkit detected')
|
||||
torch_command = os.environ.get('TORCH_COMMAND', 'torch==1.13.0a0+git6c9b55e torchvision==0.14.1a0 intel_extension_for_pytorch==1.13.120+xpu --index-url https://developer.intel.com/ipex-whl-stable-xpu')
|
||||
xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')
|
||||
@@ -222,7 +218,7 @@ def check_torch():
|
||||
try:
|
||||
import torch
|
||||
log.info(f'Torch {torch.__version__}')
|
||||
if shared.cmd_opts.use_ipex:
|
||||
if args.use_ipex:
|
||||
import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import
|
||||
log.info(f'Torch backend: Intel OneAPI {torch.__version__}')
|
||||
log.info(f'Torch detected GPU: {torch.xpu.get_device_name("xpu")} VRAM {round(torch.xpu.get_device_properties("xpu").total_memory / 1024 / 1024)}')
|
||||
@@ -373,15 +369,11 @@ def install_submodules():
|
||||
log.error(f'Error updating submodule: {submodule}')
|
||||
|
||||
|
||||
def ensure_package(pkg):
|
||||
try:
|
||||
import pkg # type: ignore
|
||||
except ImportError:
|
||||
install(pkg)
|
||||
|
||||
|
||||
def ensure_base_requirements():
|
||||
ensure_package('rich')
|
||||
try:
|
||||
import rich # pylint: disable=unused-import
|
||||
except ImportError:
|
||||
install('rich', 'rich')
|
||||
|
||||
|
||||
def install_requirements():
|
||||
|
||||
@@ -17,7 +17,6 @@ installer.parse_args()
|
||||
|
||||
import modules.cmd_args
|
||||
args, _ = modules.cmd_args.parser.parse_known_args()
|
||||
|
||||
import modules.paths_internal
|
||||
script_path = modules.paths_internal.script_path
|
||||
extensions_dir = modules.paths_internal.extensions_dir
|
||||
|
||||
+1
-2
@@ -95,7 +95,6 @@ def compatibility_args(opts, args):
|
||||
opts.dimensions_and_batch_together = True
|
||||
|
||||
group.add_argument("--lora-dir", help=argparse.SUPPRESS, default=opts.lora_dir)
|
||||
group.add_argument("--lyco-dir", help=argparse.SUPPRESS, default=opts.lyco_dir)
|
||||
args = parser.parse_args()
|
||||
if 'lyco_dir' in args: # pylint disable=unsupported-membership-test
|
||||
args.lyco_dir = opts.lyco_dir
|
||||
return args
|
||||
|
||||
@@ -13,7 +13,7 @@ startup_timer = timer.Timer()
|
||||
|
||||
import torch # pylint: disable=C0411
|
||||
try:
|
||||
import intel_extension_for_pytorch as ipex
|
||||
import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import
|
||||
except:
|
||||
pass
|
||||
import torchvision # pylint: disable=W0611,C0411
|
||||
|
||||
Reference in New Issue
Block a user