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

Merge pull request #4825 from vladmandic/feat/chroma-lora-native

Browse Source 
feat(chroma): native loader for chroma lora
This commit is contained in:
Vladimir Mandic
2026-05-05 14:39:51 +02:00
committed by GitHub
parent 6261c68396 92df94a227
commit 7f3f08ca4b
3 changed files with 551 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/lora/lora_load.py
+16 -1
View File
@@ -86,10 +86,25 @@ def load_safetensors(name, network_on_disk: network.NetworkOnDisk) -> network.Ne
if ernie_net is not None:
lora_cache[name] = ernie_net
return ernie_net
if shared.sd_model_type == 'chroma':
from pipelines.chroma import chroma_lora
lora_scale = shared.opts.extra_networks_default_multiplier
chroma_net = None
for try_fn in (chroma_lora.try_load_lora, chroma_lora.try_load_lokr, chroma_lora.try_load_loha, chroma_lora.try_load_oft):
sub = try_fn(name, network_on_disk, lora_scale)
if sub is None:
continue
if chroma_net is None:
chroma_net = sub
else:
chroma_net.modules.update(sub.modules)
if chroma_net is not None:
lora_cache[name] = chroma_net
return chroma_net
net = network.Network(name, network_on_disk)
net.mtime = os.path.getmtime(network_on_disk.filename)
state_dict = sd_models.read_state_dict(network_on_disk.filename, what='network')
if shared.sd_model_type in ['f1', 'chroma']: # if kohya flux lora, convert state_dict
if shared.sd_model_type == 'f1': # if kohya flux lora, convert state_dict
state_dict = lora_convert._convert_kohya_flux_lora_to_diffusers(state_dict) or state_dict # pylint: disable=protected-access
if shared.sd_model_type == 'sd3': # if kohya flux lora, convert state_dict
try: