1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add onnx to installer

pull/3016/head
Vladimir Mandic 2024-03-26 17:05:11 -04:00
parent 1227e76128
commit 738f115d06
3 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CHANGELOG.md
View File

@ -6,6 +6,7 @@
- Quick apply style - Quick apply style
- SC LoRA - SC LoRA
- DoRA - DoRA
- SDXS VAE: https://huggingface.co/IDKiro/sdxs-512-0.9
- Control API/CLI - Control API/CLI
- scripts - scripts
- units - units

2
installer.py
View File

@ -390,6 +390,8 @@ def check_python():
# check onnx version # check onnx version
def check_onnx(): def check_onnx():
if not installed('onnx', quiet=True):
install('onnx', 'onnx', ignore=True)
if not installed('onnxruntime', quiet=True) and not installed('onnxruntime-gpu', quiet=True): # allow either if not installed('onnxruntime', quiet=True) and not installed('onnxruntime-gpu', quiet=True): # allow either
install('onnxruntime', 'onnxruntime', ignore=True) install('onnxruntime', 'onnxruntime', ignore=True)

1
modules/processing.py
View File

@ -403,6 +403,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
images.save_image(grid, p.outpath_grids, "", p.all_seeds[0], p.all_prompts[0], shared.opts.grid_format, info=infotext(-1), p=p, grid=True, suffix="-grid") # main save grid images.save_image(grid, p.outpath_grids, "", p.all_seeds[0], p.all_prompts[0], shared.opts.grid_format, info=infotext(-1), p=p, grid=True, suffix="-grid") # main save grid
if shared.backend == shared.Backend.DIFFUSERS: if shared.backend == shared.Backend.DIFFUSERS:
from modules import ipadapter
ipadapter.unapply(shared.sd_model) ipadapter.unapply(shared.sd_model)
if not p.disable_extra_networks: if not p.disable_extra_networks: