fix(api): add missing arguments to prompt patch (#336)

ssube · Apr 15, 2023 · dd77166 · dd77166
1 parent b076326
commit dd77166
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
@@ -35,6 +35,8 @@ def expand_prompt(
     num_images_per_prompt: int,
     do_classifier_free_guidance: bool,
     negative_prompt: Optional[str] = None,
+    prompt_embeds: Optional[np.ndarray] = None,
+    negative_prompt_embeds: Optional[np.ndarray] = None,
     skip_clip_states: Optional[int] = 0,
 ) -> "np.NDArray":
     # self provides: