fix(api): add base token for Textual Inversion as sum layer
This commit is contained in:
parent
aefa5b4613
commit
19d4d554c3
|
@ -92,7 +92,12 @@ def blend_textual_inversions(
|
||||||
else:
|
else:
|
||||||
embeds[token] = layer
|
embeds[token] = layer
|
||||||
|
|
||||||
# add sum layer to embeds
|
# add base and sum tokens to embeds
|
||||||
|
if base_token in embeds:
|
||||||
|
embeds[base_token] += sum_layer
|
||||||
|
else:
|
||||||
|
embeds[base_token] = sum_layer
|
||||||
|
|
||||||
sum_token = f"{base_token}-all"
|
sum_token = f"{base_token}-all"
|
||||||
if sum_token in embeds:
|
if sum_token in embeds:
|
||||||
embeds[sum_token] += sum_layer
|
embeds[sum_token] += sum_layer
|
||||||
|
|
Loading…
Reference in New Issue