From f28be14639e37db890050d479d9a511ce0a40704 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 2 Jul 2026 01:07:36 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/training/diffusion_lora_trainer.py | 9 ++++++--- studio/backend/tests/test_diffusion_lora_trainer.py | 4 +--- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/studio/backend/core/training/diffusion_lora_trainer.py b/studio/backend/core/training/diffusion_lora_trainer.py index fba01d0b14..70f7c8aedf 100644 --- a/studio/backend/core/training/diffusion_lora_trainer.py +++ b/studio/backend/core/training/diffusion_lora_trainer.py @@ -282,7 +282,6 @@ def _assert_trusted_base_model(base_model: str) -> None: BEFORE ``from_pretrained`` so an untrusted remote repo (which could ship pickle weights) is never fetched or deserialised.""" from core.inference.diffusion import _is_trusted_diffusion_repo - if not _is_trusted_diffusion_repo(base_model): raise ValueError( f"Refusing to train from untrusted base model '{base_model}'. Use a local path or " @@ -351,8 +350,12 @@ def run_diffusion_lora_training( if _check_stop(): out_dir = Path(cfg.output_dir).expanduser() _emit( - on_event, "complete", - output_dir = str(out_dir), lora_path = None, stopped = True, steps_run = 0, + on_event, + "complete", + output_dir = str(out_dir), + lora_path = None, + stopped = True, + steps_run = 0, ) return str(out_dir) diff --git a/studio/backend/tests/test_diffusion_lora_trainer.py b/studio/backend/tests/test_diffusion_lora_trainer.py index d35e1b3f1a..2f83eee1d3 100644 --- a/studio/backend/tests/test_diffusion_lora_trainer.py +++ b/studio/backend/tests/test_diffusion_lora_trainer.py @@ -123,9 +123,7 @@ def test_config_from_dict_ignores_unknown_and_tuples_targets(): def test_config_rejects_zero_lora_alpha(): # An explicit zero alpha would scale the adapter to nothing; reject it. with pytest.raises(ValueError, match = "lora_alpha"): - DiffusionLoraConfig( - base_model = "b", data_dir = "d", output_dir = "o", lora_alpha = 0 - ).normalized() + DiffusionLoraConfig(base_model = "b", data_dir = "d", output_dir = "o", lora_alpha = 0).normalized() def test_config_coerces_string_learning_rate():