mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
add sefi placeholder
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -137,6 +137,8 @@ def get_model_type(pipe):
|
||||
model_type = 'hunyuanimage'
|
||||
elif 'sdxs-1b' in name:
|
||||
model_type = 'sdxs'
|
||||
elif 'SeFi' in name:
|
||||
model_type = 'sefi'
|
||||
# video models
|
||||
elif "Kandinsky5" in name and '2V' in name:
|
||||
model_type = 'kandinsky5video'
|
||||
|
||||
@@ -173,6 +173,8 @@ def guess_by_name(fn, current_guess):
|
||||
new_guess = 'VIBE'
|
||||
elif 'joyai-image-edit' in fn.lower() or 'joy-image-edit' in fn.lower():
|
||||
new_guess = 'JoyEdit'
|
||||
elif 'sefi-image' in fn.lower():
|
||||
new_guess = 'SeFi'
|
||||
if debug_load:
|
||||
log.trace(f'Autodetect: method=name file="{fn}" previous="{current_guess}" current="{new_guess}"')
|
||||
return new_guess or current_guess
|
||||
|
||||
@@ -602,6 +602,10 @@ def load_diffuser_force(detected_model_type: str, checkpoint_info: CheckpointInf
|
||||
from pipelines.model_sdxs import load_sdxs
|
||||
sd_model = load_sdxs(checkpoint_info, diffusers_load_config)
|
||||
allow_post_quant = False
|
||||
elif model_type in ['SeFi']:
|
||||
from pipelines.model_sefi import load_sefi
|
||||
sd_model = load_sefi(checkpoint_info, diffusers_load_config)
|
||||
allow_post_quant = False
|
||||
except Exception as e:
|
||||
log.error(f'Load {op}: path="{checkpoint_info.path}" {e}')
|
||||
errors.display(e, 'Load')
|
||||
|
||||
Reference in New Issue
Block a user