Skip to content

Commit b8b73d8

Browse files
committed
fix(api): make sure UNet wrapper always has embeds defined
1 parent bc946bb commit b8b73d8

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

api/onnx_web/diffusers/load.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -402,6 +402,11 @@ def optimize_pipeline(
402402

403403

404404
class UNetWrapper(object):
405+
prompt_embeds: Optional[List[np.ndarray]] = None
406+
prompt_index: int = 0
407+
server: ServerContext
408+
wrapped: OnnxRuntimeModel
409+
405410
def __init__(
406411
self,
407412
server: ServerContext,

0 commit comments

Comments
 (0)