Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

add bria-3.2 model support

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-07-23 10:42:39 -04:00
parent 444974a6ff
commit 32e1787b47
14 changed files with 2075 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/modeldata.py
+2
View File
@@ -56,6 +56,8 @@ def get_model_type(pipe):
model_type = 'pixartsigma'
elif "PixArtAlpha" in name:
model_type = 'pixartalpha'
elif "Bria" in name:
model_type = 'bria'
# video models
elif "CogVideo" in name:
model_type = 'cogvideo'
modules/sd_detect.py
+2
View File
@@ -99,6 +99,8 @@ def guess_by_name(fn, current_guess):
return 'FLite'
elif 'wan' in fn.lower():
return 'WanAI'
elif 'bria' in fn.lower():
return 'Bria'
return current_guess
modules/sd_models.py
+4
View File
@@ -369,6 +369,10 @@ def load_diffuser_force(model_type, checkpoint_info, diffusers_load_config, op='
from pipelines.model_wanai import load_wan
sd_model = load_wan(checkpoint_info, diffusers_load_config)
allow_post_quant = False
elif model_type in ['Bria']:
from pipelines.model_bria import load_bria
sd_model = load_bria(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:
modules/shared_items.py
+6 -5
View File
@@ -47,12 +47,13 @@ pipelines = {
'WanAI': getattr(diffusers, 'WanPipeline', None),
# dynamically imported and redefined later
'Meissonic': getattr(diffusers, 'DiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser
'Monetico': getattr(diffusers, 'DiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser
'OmniGen2': getattr(diffusers, 'DiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser
'InstaFlow': getattr(diffusers, 'DiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser
'SegMoE': getattr(diffusers, 'DiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser
'Meissonic': getattr(diffusers, 'DiffusionPipeline', None),
'Monetico': getattr(diffusers, 'DiffusionPipeline', None),
'OmniGen2': getattr(diffusers, 'DiffusionPipeline', None),
'InstaFlow': getattr(diffusers, 'DiffusionPipeline', None),
'SegMoE': getattr(diffusers, 'DiffusionPipeline', None),
'FLite': getattr(diffusers, 'DiffusionPipeline', None),
'Bria': getattr(diffusers, 'DiffusionPipeline', None),
}
initialize_onnx()