fix typo
This commit is contained in:
parent
5da3cfa26d
commit
504c1ce084
|
@ -21,7 +21,7 @@ def wrap_encoder(text_encoder):
|
||||||
self, token_ids, attention_mask, output_hidden_states=True, return_dict=True
|
self, token_ids, attention_mask, output_hidden_states=True, return_dict=True
|
||||||
):
|
):
|
||||||
# TODO: does compel use attention masks?
|
# 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 return_dict:
|
||||||
if output_hidden_states:
|
if output_hidden_states:
|
||||||
hidden_states = outputs[2:]
|
hidden_states = outputs[2:]
|
||||||
|
|
Loading…
Reference in New Issue