mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
add step1x-edit and vibe-edit
Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
@@ -453,6 +453,14 @@ def load_diffuser_force(detected_model_type, checkpoint_info, diffusers_load_con
|
||||
from pipelines.model_bria import load_bria
|
||||
sd_model = load_bria(checkpoint_info, diffusers_load_config)
|
||||
allow_post_quant = False
|
||||
elif model_type in ['Step1X-Edit']:
|
||||
from pipelines.model_step1x_edit import load_step1x_edit
|
||||
sd_model = load_step1x_edit(checkpoint_info, diffusers_load_config)
|
||||
allow_post_quant = False
|
||||
elif model_type in ['VIBE']:
|
||||
from pipelines.model_vibe import load_vibe
|
||||
sd_model = load_vibe(checkpoint_info, diffusers_load_config)
|
||||
allow_post_quant = False
|
||||
elif model_type in ['Qwen']:
|
||||
from pipelines.model_qwen import load_qwen
|
||||
sd_model = load_qwen(checkpoint_info, diffusers_load_config)
|
||||
|
||||
Reference in New Issue
Block a user