fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-06 18:08:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add pre_check() to settings component to load autoload models

Browse Source
This commit is contained in:
henryruhs 2023-08-05 21:07:04 +02:00
parent cf8b52acaf
commit 3981ef619b
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
roop/uis/__components__/settings.py
View File

@ -3,7 +3,7 @@ import gradio
import onnxruntime
import roop.globals
from roop.processors.frame.core import list_frame_processors_names, clear_frame_processors_modules
from roop.processors.frame.core import list_frame_processors_names, load_frame_processor_module, clear_frame_processors_modules
from roop.uis import core as ui
@ -60,10 +60,13 @@ def render() -> None:
def update_frame_processors(frame_processors: List[str]) -> Dict[Any, Any]:
clear_frame_processors_modules()
roop.globals.frame_processors = frame_processors
for frame_processor in roop.globals.frame_processors:
frame_processor_module = load_frame_processor_module(frame_processor)
frame_processor_module.pre_check()
return gradio.update(value=frame_processors, choices=sort_frame_processors(frame_processors))
def sort_frame_processors(frame_processors: List[str]):
def sort_frame_processors(frame_processors: List[str]) -> list[str]:
frame_processor_key = lambda frame_processor: frame_processors.index(frame_processor) if frame_processor in frame_processors else len(frame_processors)
return sorted(list_frame_processors_names(), key=frame_processor_key)