From 44a20a47c1e67444c3fe705cfa5eb656a5727aab Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 9 Aug 2025 12:15:36 -0400 Subject: [PATCH] update todo Signed-off-by: Vladimir Mandic --- TODO.md | 7 ++++--- cli/test-all-models.py | 33 ++++++++++++++++++++++++++++++ extensions-builtin/sdnext-modernui | 2 +- modules/ui_postprocessing.py | 6 +++--- 4 files changed, 41 insertions(+), 7 deletions(-) create mode 100644 cli/test-all-models.py diff --git a/TODO.md b/TODO.md index 4a8f601f2..da9883bd8 100644 --- a/TODO.md +++ b/TODO.md @@ -79,11 +79,12 @@ Main ToDo list can be found at [GitHub projects](https://github.com/users/vladma - lora: add other quantization types - lora: add t5 key support for sd35/f1 - lora: maybe force imediate quantization -- lora: support pre-quantized flux -- model load: cogview4: balanced offload does not work for GlmModel -- model load: add ChromaFillPipeline, ChromaControlPipeline, ChromaImg2ImgPipeline etc when available +- model load: add ChromaControlPipeline, ChromaInpaintPipeline +- model load: cogview4 balanced offload does not work for GlmModel - model load: force-reloading entire model as loading transformers only leads to massive memory usage +- model load: group offload - model load: implement model in-memory caching - modernui: monkey-patch for missing tabs.select event +- modules/lora/lora_extract.py:188:9: W0511: TODO: lora: support pre-quantized flux - processing: remove duplicate mask params - resize image: enable full VAE mode for resize-latent diff --git a/cli/test-all-models.py b/cli/test-all-models.py new file mode 100644 index 000000000..6bc768317 --- /dev/null +++ b/cli/test-all-models.py @@ -0,0 +1,33 @@ +models = [ + "sd_xl_base_1.0", + "tempestByVlad_baseV01", + "huggingface/stabilityai/stable-cascade", + "stabilityai/stable-diffusion-3.5-medium", + "stabilityai/stable-diffusion-3.5-large", + "black-forest-labs/FLUX.1-dev", + "black-forest-labs/FLUX.1-Kontext-dev", + "black-forest-labs/FLUX.1-Krea-dev", + "vladmandic/chroma-unlocked-v50", + "vladmandic/chroma-unlocked-v50-annealed", + "Qwen/Qwen-Image", + "ostris/Flex.2-preview", + "Wan-AI/Wan2.1-T2V-1.3B-Diffusers", + "Wan-AI/Wan2.1-T2V-14B-Diffusers", + "Freepik/F-Lite", + "Freepik/F-Lite-Texture", + "Efficient-Large-Model/SANA1.5_1.6B_1024px_diffusers", + "Efficient-Large-Model/SANA1.5_4.8B_1024px_diffusers", + "nvidia/Cosmos-Predict2-2B-Text2Image", + "nvidia/Cosmos-Predict2-14B-Text2Image", + "OmniGen2/OmniGen2", + "fal/AuraFlow-v0.3", + "PixArt-alpha/PixArt-XL-2-1024-MS", + "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS", + "Tencent-Hunyuan/HunyuanDiT-v1.2-Diffusers", + "Alpha-VLLM/Lumina-Next-SFT-diffusers", + "Alpha-VLLM/Lumina-Image-2.0", + "HiDream-ai/HiDream-I1-Dev", + "HiDream-ai/HiDream-I1-Full", + "Kwai-Kolors/Kolors-diffusers", + "briaai/BRIA-3.2", +] diff --git a/extensions-builtin/sdnext-modernui b/extensions-builtin/sdnext-modernui index 3a2d19880..5ec177ccb 160000 --- a/extensions-builtin/sdnext-modernui +++ b/extensions-builtin/sdnext-modernui @@ -1 +1 @@ -Subproject commit 3a2d1988047e0880991468313292c057742b139e +Subproject commit 5ec177ccbf818cc759bd5306f450b7bcdaaf2744 diff --git a/modules/ui_postprocessing.py b/modules/ui_postprocessing.py index 24f787259..b575fa67a 100644 --- a/modules/ui_postprocessing.py +++ b/modules/ui_postprocessing.py @@ -38,11 +38,11 @@ def create_ui(): with gr.Row(elem_id=f"{id_part}_generate_box", elem_classes="generate-box"): submit = gr.Button('Generate', elem_id=f"{id_part}_generate", variant='primary') interrupt = gr.Button('Stop', elem_id=f"{id_part}_interrupt", variant='secondary') - interrupt.click(fn=lambda: shared.state.interrupt(), inputs=[], outputs=[]) + interrupt.click(fn=shared.state.interrupt, inputs=[], outputs=[]) skip = gr.Button('Skip', elem_id=f"{id_part}_skip", variant='secondary') - skip.click(fn=lambda: shared.state.skip(), inputs=[], outputs=[]) + skip.click(fn=shared.state.skip, inputs=[], outputs=[]) pause = gr.Button('Pause', elem_id=f"{id_part}_pause") - pause.click(fn=lambda: shared.state.pause(), _js='checkPaused', inputs=[], outputs=[]) + pause.click(fn=shared.state.pause, _js='checkPaused', inputs=[], outputs=[]) result_images, generation_info, html_info, html_info_formatted, html_log = ui_common.create_output_panel("extras") gr.HTML('File metadata') exif_info = gr.HTML(elem_id="pnginfo_html_info")