hyimage placeholder

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-09-10 10:09:56 -04:00
parent ca2e5ae610
commit 4c684143a4
4 changed files with 51 additions and 0 deletions
+2
View File
@@ -113,6 +113,8 @@ def guess_by_name(fn, current_guess):
return 'Kandinsky 2.2'
elif 'kandinsky-3' in fn.lower():
return 'Kandinsky 3.0'
elif 'hunyuanimage' in fn.lower():
return 'HunyuanImage'
return current_guess
+4
View File
@@ -394,6 +394,10 @@ def load_diffuser_force(model_type, checkpoint_info, diffusers_load_config, op='
from pipelines.model_hdm import load_hdm
sd_model = load_hdm(checkpoint_info, diffusers_load_config)
allow_post_quant = False
elif model_type in ['HunyuanImage']:
from pipelines.model_hyimage import load_hyimage
sd_model = load_hyimage(checkpoint_info, diffusers_load_config)
allow_post_quant = False
except Exception as e:
shared.log.error(f'Load {op}: path="{checkpoint_info.path}" {e}')
if debug_load:
+1
View File
@@ -55,6 +55,7 @@ pipelines = {
'FLite': getattr(diffusers, 'DiffusionPipeline', None),
'Bria': getattr(diffusers, 'DiffusionPipeline', None),
'hdm': getattr(diffusers, 'DiffusionPipeline', None),
'HunyuanImage': getattr(diffusers, 'DiffusionPipeline', None),
}