Skip to content

Commit 3d8b1af

Browse files
Support string_to_param nested dict
format: bundle_emb.EMBNAME.string_to_param.KEYNAME
1 parent 2aa485b commit 3d8b1af

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

extensions-builtin/Lora/networks.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,11 @@ def load_network(name, network_on_disk):
157157
if key_network_without_network_parts == "bundle_emb":
158158
emb_name, vec_name = network_part.split(".", 1)
159159
emb_dict = bundle_embeddings.get(emb_name, {})
160-
emb_dict[vec_name] = weight
160+
if vec_name.split('.')[0] == 'string_to_param':
161+
_, k2 = vec_name.split('.', 1)
162+
emb_dict['string_to_param'] = {k2: weight}
163+
else:
164+
emb_dict[vec_name] = weight
161165
bundle_embeddings[emb_name] = emb_dict
162166

163167
key = convert_diffusers_name_to_compvis(key_network_without_network_parts, is_sd2)
@@ -301,6 +305,7 @@ def load_networks(names, te_multipliers=None, unet_multipliers=None, dyn_dims=No
301305

302306
if emb_db.expected_shape == -1 or emb_db.expected_shape == embedding.shape:
303307
emb_db.register_embedding(embedding, shared.sd_model)
308+
print(f'registered bundle embedding: {embedding.name}')
304309
else:
305310
emb_db.skipped_embeddings[name] = embedding
306311

0 commit comments

Comments
 (0)