From 19d4d554c3f382ee1cee7c104d24c6f7f67d8dc3 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 19 Mar 2023 18:04:21 -0500 Subject: [PATCH] fix(api): add base token for Textual Inversion as sum layer --- api/onnx_web/convert/diffusion/textual_inversion.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/convert/diffusion/textual_inversion.py b/api/onnx_web/convert/diffusion/textual_inversion.py index ec02f603..51f0978c 100644 --- a/api/onnx_web/convert/diffusion/textual_inversion.py +++ b/api/onnx_web/convert/diffusion/textual_inversion.py @@ -92,7 +92,12 @@ def blend_textual_inversions( else: 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" if sum_token in embeds: embeds[sum_token] += sum_layer