mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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),
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user