Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix hidream-o1 loader

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-05-14 21:08:08 +02:00
parent 74e433caa7
commit b9d92a7fef
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pipelines/model_hidream.py
+7 -2
View File
@@ -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,
)