diff --git a/scripts/script.py b/scripts/script.py index 5a64b1c..3b3d227 100644 --- a/scripts/script.py +++ b/scripts/script.py @@ -237,7 +237,8 @@ def on_ui_tabs(): elem_id=tab_prefix + "preview", visible=config.show_image_preview, readonly=True, - ), columns=4) + columns=4, + ) def on_refresh(image=False, show_images=config.show_image_preview): cid = f"Current ID: {horde.state.id}" diff --git a/stable_horde/horde.py b/stable_horde/horde.py index 12adde1..85a8db2 100644 --- a/stable_horde/horde.py +++ b/stable_horde/horde.py @@ -263,9 +263,7 @@ class StableHorde: ), SamplerData( "LMS Karras", - lambda model, funcname="sample_lms": KDiffusionSampler( - funcname, model - ), + lambda model, funcname="sample_lms": KDiffusionSampler(funcname, model), ["k_lms_ka"], {"scheduler": "karras"}, ),