fix(api): make sure UNet wrapper always has embeds defined

ssube · Apr 27, 2023 · b8b73d8 · b8b73d8
1 parent bc946bb
commit b8b73d8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
@@ -402,6 +402,11 @@ def optimize_pipeline(
 
 
 class UNetWrapper(object):
+    prompt_embeds: Optional[List[np.ndarray]] = None
+    prompt_index: int = 0
+    server: ServerContext
+    wrapped: OnnxRuntimeModel
+
     def __init__(
         self,
         server: ServerContext,