fix(api): use padding token for prompt padding

ssube · Apr 24, 2023 · 2fe6a87 · 2fe6a87
1 parent 3f4dd70
commit 2fe6a87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
@@ -200,7 +200,7 @@ def expand_prompt(
             negative_prompt_embeds,
             [(0, 0), (0, negative_padding), (0, 0)],
             mode="constant",
-            constant_values=0,
+            constant_values=self.tokenizer.pad_token_id,
         )
         negative_prompt_embeds = np.repeat(
             negative_prompt_embeds, num_images_per_prompt, axis=0