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

ONNX/Olive support .safetensors VAE.

Browse Source
This commit is contained in:
Seunghoon Lee
2024-01-14 17:23:02 +09:00
parent 0fac69cbf5
commit ea2ac32cf2
6 changed files with 282 additions and 230 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ui.py
+7
View File
@@ -368,6 +368,13 @@ def create_ui(startup_timer = None):
interfaces += [(interrogate_interface, "Interrogate", "interrogate")]
interfaces += [(train_interface, "Train", "train")]
interfaces += [(models_interface, "Models", "models")]
if shared.opts.onnx_show_menu:
with gr.Blocks(analytics_enabled=False) as onnx_interface:
if shared.backend == shared.Backend.DIFFUSERS:
from modules import ui_onnx
ui_onnx.create_ui()
timer.startup.record("ui-onnx")
interfaces += [(onnx_interface, "ONNX", "onnx")]
interfaces += script_callbacks.ui_tabs_callback()
interfaces += [(settings_interface, "System", "system")]