From 56d714368ad7b9de11219061e8e49d7d4dab9ce3 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 16 Jun 2025 11:44:22 +0200 Subject: [PATCH] Fix UI --- facefusion/uis/components/benchmark.py | 3 ++- facefusion/uis/components/benchmark_options.py | 10 +++++----- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/facefusion/uis/components/benchmark.py b/facefusion/uis/components/benchmark.py index 55e1577..89939a6 100644 --- a/facefusion/uis/components/benchmark.py +++ b/facefusion/uis/components/benchmark.py @@ -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 ] diff --git a/facefusion/uis/components/benchmark_options.py b/facefusion/uis/components/benchmark_options.py index a7f07cf..60322af 100644 --- a/facefusion/uis/components/benchmark_options.py +++ b/facefusion/uis/components/benchmark_options.py @@ -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)