jones/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix UI

Browse Source
This commit is contained in:
henryruhs
2025-06-16 11:44:22 +02:00
parent 0feecfd9d7
commit 56d714368a
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
facefusion/uis/components/benchmark.py
View File

@@ -56,4 +56,5 @@ def start(benchmark_resolutions : List[str], benchmark_cycles : int) -> Generato
state_manager.sync_item('execution_thread_count')
state_manager.sync_item('execution_queue_count')
yield from benchmarker.run()
for benchmark in benchmarker.run():
yield [ list(benchmark_set.values()) for benchmark_set in benchmark ]

10
facefusion/uis/components/benchmark_options.py
View File

@@ -3,7 +3,7 @@ from typing import Optional
import gradio
from facefusion import wording
from facefusion.benchmarker import BENCHMARKS
import facefusion.choices
from facefusion.uis.core import register_ui_component
BENCHMARK_RESOLUTIONS_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None
@@ -16,15 +16,15 @@ def render() -> None:
BENCHMARK_RESOLUTIONS_CHECKBOX_GROUP = gradio.CheckboxGroup(
label = wording.get('uis.benchmark_resolutions_checkbox_group'),
choices = list(BENCHMARKS.keys()),
value = list(BENCHMARKS.keys())
choices = facefusion.choices.benchmark_resolutions,
value = facefusion.choices.benchmark_resolutions
)
BENCHMARK_CYCLES_SLIDER = gradio.Slider(
label = wording.get('uis.benchmark_cycles_slider'),
value = 5,
step = 1,
minimum = 1,
maximum = 10
minimum = min(facefusion.choices.benchmark_cycles_range),
maximum = max(facefusion.choices.benchmark_cycles_range)
)
register_ui_component('benchmark_resolutions_checkbox_group', BENCHMARK_RESOLUTIONS_CHECKBOX_GROUP)
register_ui_component('benchmark_cycles_slider', BENCHMARK_CYCLES_SLIDER)