From a85009ad3609431c9abb75054fdbcd3533a64764 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 14 May 2026 06:37:15 +0200 Subject: [PATCH 1/2] fix hidream prequant loader Signed-off-by: Vladimir Mandic --- pipelines/model_hidream.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pipelines/model_hidream.py b/pipelines/model_hidream.py index 7d1fcd5e4..a0875674d 100644 --- a/pipelines/model_hidream.py +++ b/pipelines/model_hidream.py @@ -51,6 +51,7 @@ def load_hidream_o1(checkpoint_info, diffusers_load_config=None): repo_id, cache_dir=shared.opts.hfcache_dir, trust_remote_code=True, + subfolder='transformer' if 'vladmandic' in repo_id.lower() else None, **load_args, **quant_args, ) @@ -59,6 +60,7 @@ def load_hidream_o1(checkpoint_info, diffusers_load_config=None): processor = transformers.AutoProcessor.from_pretrained( repo_id, + subfolder='processor' if 'vladmandic' in repo_id.lower() else None, cache_dir=shared.opts.hfcache_dir, trust_remote_code=True, ) From dedb130b2c15ef9d91f9593213ccffcb70d98be4 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 14 May 2026 21:08:08 +0200 Subject: [PATCH 2/2] fix hidream-o1 loader Signed-off-by: Vladimir Mandic --- pipelines/model_hidream.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/pipelines/model_hidream.py b/pipelines/model_hidream.py index a0875674d..8301f4f0c 100644 --- a/pipelines/model_hidream.py +++ b/pipelines/model_hidream.py @@ -47,20 +47,25 @@ def load_hidream_o1(checkpoint_info, diffusers_load_config=None): o1_load_config = diffusers_load_config.copy() o1_load_config['trust_remote_code'] = True + path_args = {} + if 'vladmandic' in repo_id.lower(): + path_args['subfolder'] = 'transformer' transformer = HiDreamO1Qwen3VLTransformer.from_pretrained( repo_id, cache_dir=shared.opts.hfcache_dir, trust_remote_code=True, - subfolder='transformer' if 'vladmandic' in repo_id.lower() else None, + **path_args, **load_args, **quant_args, ) if shared.opts.diffusers_offload_mode != 'none' and transformer is not None: sd_models.move_model(transformer, devices.cpu) + if 'vladmandic' in repo_id.lower(): + path_args['subfolder'] = 'processor' processor = transformers.AutoProcessor.from_pretrained( repo_id, - subfolder='processor' if 'vladmandic' in repo_id.lower() else None, + **path_args, cache_dir=shared.opts.hfcache_dir, trust_remote_code=True, )