apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): add base token for Textual Inversion as sum layer

This commit is contained in:
Sean Sube 2023-03-19 18:04:21 -05:00
parent aefa5b4613
commit 19d4d554c3
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/onnx_web/convert/diffusion/textual_inversion.py
View File

@ -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