From 504c1ce084eb43623a55efa9fd43d113b353e6cb Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 2 Mar 2024 22:10:20 -0600 Subject: [PATCH] fix typo --- api/onnx_web/prompt/compel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/prompt/compel.py b/api/onnx_web/prompt/compel.py index d13f4c56..634a4ff2 100644 --- a/api/onnx_web/prompt/compel.py +++ b/api/onnx_web/prompt/compel.py @@ -21,7 +21,7 @@ def wrap_encoder(text_encoder): self, token_ids, attention_mask, output_hidden_states=True, return_dict=True ): # TODO: does compel use attention masks? - outputs = text_encoder(inputs_ids=token_ids.numpy().astype(np.int32)) + outputs = text_encoder(input_ids=token_ids.numpy().astype(np.int32)) if return_dict: if output_hidden_states: hidden_states = outputs[2:]