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

add LibreFlux segmentation controlnet for FLUX.1

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-10-28 09:18:56 -04:00
parent 3e996eed27
commit 6f05ae71cd
2 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/control/units/controlnet.py
+15 -1
View File
@@ -89,7 +89,8 @@ predefined_f1 = {
"Shakker-Labs Depth F1": 'Shakker-Labs/FLUX.1-dev-ControlNet-Depth',
"XLabs-AI Canny F1": 'XLabs-AI/flux-controlnet-canny-diffusers',
"XLabs-AI Depth F1": 'XLabs-AI/flux-controlnet-depth-diffusers',
"XLabs-AI HED F1": 'XLabs-AI/flux-controlnet-hed-diffusers'
"XLabs-AI HED F1": 'XLabs-AI/flux-controlnet-hed-diffusers',
"LibreFlux F1": 'neuralvfx/LibreFlux-ControlNet',
}
predefined_sd3 = {
"StabilityAI Canny SD35": 'diffusers-internal-dev/sd35-controlnet-canny-8b',
@@ -119,6 +120,15 @@ variants = {
'NoobAI SoftEdge XL': 'fp16',
'TTPlanet Tile Realistic XL': 'fp16',
}
subfolders = {
"LibreFlux F1": 'controlnet',
}
remote_code = {
"LibreFlux F1": True,
}
models = {}
all_models = {}
all_models.update(predefined_sd15)
@@ -326,6 +336,10 @@ class ControlNet():
self.load_config['use_safetensors'] = True
if variants.get(model_id, None) is not None:
kwargs['variant'] = variants[model_id]
if subfolders.get(model_id, None) is not None:
self.load_config['subfolder'] = subfolders[model_id]
if remote_code.get(model_id, None) is not None:
self.load_config['trust_remote_code'] = remote_code[model_id]
try:
self.model = cls.from_pretrained(model_path, **self.load_config, **kwargs)
except Exception as e: