apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix typo

This commit is contained in:
Sean Sube 2024-03-02 22:10:20 -06:00
parent 5da3cfa26d
commit 504c1ce084
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/prompt/compel.py
View File

@ -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:]