Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

cache diffusers version

Browse Source
This commit is contained in:
Vladimir Mandic
2024-09-07 12:12:45 -04:00
parent 084ddd49c6
commit 8f0d08e9a4
2 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+3 -1
View File
@@ -56,6 +56,7 @@ args = Dot({
'uv': False,
})
git_commit = "unknown"
diffusers_commit = "unknown"
submodules_commit = {
'sd-webui-controlnet': 'ecd33eb',
# 'stable-diffusion-webui-images-browser': '27fe4a7',
@@ -446,7 +447,8 @@ def check_diffusers():
if minor > 0:
pip('uninstall --yes diffusers', ignore=True, quiet=True, uv=False)
pip(f'install --upgrade git+https://github.com/huggingface/diffusers@{sha}', ignore=False, quiet=True, uv=False)
opts['diffusers_version'] = sha
global diffusers_commit # pylint: disable=global-statement
diffusers_commit = sha
# check onnx version
webui.py
+8 -6
View File
@@ -11,7 +11,7 @@ from threading import Thread
import modules.loader
import torch # pylint: disable=wrong-import-order
from modules import timer, errors, paths # pylint: disable=unused-import
from installer import log, git_commit, custom_excepthook
from installer import log, git_commit, diffusers_commit, custom_excepthook
import ldm.modules.encoders.modules # pylint: disable=unused-import, wrong-import-order
from modules import shared, extensions, gr_tempdir, modelloader # pylint: disable=ungrouped-imports
from modules import extra_networks, ui_extra_networks # pylint: disable=ungrouped-imports
@@ -64,6 +64,7 @@ import modules.sd_hijack
timer.startup.record("ldm")
modules.loader.initialized = True
def check_rollback_vae():
if shared.cmd_opts.rollback_vae:
if not torch.cuda.is_available():
@@ -77,7 +78,6 @@ def check_rollback_vae():
shared.cmd_opts.rollback_vae = False
def initialize():
log.debug('Initializing')
check_rollback_vae()
@@ -102,10 +102,10 @@ def initialize():
timer.startup.record("models")
import modules.postprocess.codeformer_model as codeformer
codeformer.setup_model(opts.codeformer_models_path)
codeformer.setup_model(shared.opts.codeformer_models_path)
sys.modules["modules.codeformer_model"] = codeformer
import modules.postprocess.gfpgan_model as gfpgan
gfpgan.setup_model(opts.gfpgan_models_path)
gfpgan.setup_model(shared.opts.gfpgan_models_path)
timer.startup.record("face-restore")
log.debug('Load extensions')
@@ -159,7 +159,7 @@ def initialize():
def load_model():
modules.devices.set_cuda_params()
if not opts.sd_checkpoint_autoload or (shared.cmd_opts.ckpt is not None and shared.cmd_opts.ckpt.lower() != 'none'):
if not shared.opts.sd_checkpoint_autoload or (shared.cmd_opts.ckpt is not None and shared.cmd_opts.ckpt.lower() != 'none'):
log.debug('Model auto load disabled')
else:
shared.state.begin('Load')
@@ -217,9 +217,11 @@ def start_common():
log.info(f'Using data path: {shared.cmd_opts.data_dir}')
if shared.cmd_opts.models_dir is not None and len(shared.cmd_opts.models_dir) > 0 and shared.cmd_opts.models_dir != 'models':
log.info(f'Using models path: {shared.cmd_opts.models_dir}')
create_paths(opts)
create_paths(shared.opts)
async_policy()
initialize()
if diffusers_commit != 'unknown':
shared.opts.diffusers_version = diffusers_commit # update installed diffusers version
if shared.opts.clean_temp_dir_at_start:
gr_tempdir.cleanup_tmpdr()
timer.startup.record("cleanup")