From 3533258980f393bcb3e44fff2adbacd7d3b9587e Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Fri, 11 Apr 2025 22:32:57 -0400 Subject: [PATCH] hidream whitelist samplers Signed-off-by: Vladimir Mandic --- modules/sd_samplers.py | 2 +- modules/shared.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index be375dbff..f73520027 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -12,7 +12,7 @@ samplers = all_samplers samplers_for_img2img = all_samplers samplers_map = {} loaded_config = None -flow_models = ['Flux', 'StableDiffusion3', 'Lumina', 'AuraFlow', 'Sana', 'CogView4'] +flow_models = ['Flux', 'StableDiffusion3', 'Lumina', 'AuraFlow', 'Sana', 'CogView4', 'HiDream'] flow_models += ['Hunyuan', 'LTX', 'Mochi'] diff --git a/modules/shared.py b/modules/shared.py index 72b2e0994..9a2eb8cb4 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -415,8 +415,8 @@ options_templates.update(options_section(('sd', "Models & Loading"), { })) options_templates.update(options_section(('model_options', "Models Options"), { - "model_sd3_disable_te5": OptionInfo(False, "SD3 disable T5 encoder"), - "model_h1_llama_repo": OptionInfo("meta-llama/Meta-Llama-3.1-8B-Instruct", "LLama repo", gr.Textbox), + "model_sd3_disable_te5": OptionInfo(False, "StableDiffusion3: T5 disable encoder"), + "model_h1_llama_repo": OptionInfo("meta-llama/Meta-Llama-3.1-8B-Instruct", "HiDream: LLama repo", gr.Textbox), })) options_templates.update(options_section(('vae_encoder', "Variable Auto Encoder"), {