transformers
CLI: Don't check the model head when there is no model head
#18733
Merged
Merged
CLI: Don't check the model head when there is no model head #18733
gante
merged
2 commits into
main
from
pt_tf_cli_automodel_fix
Don't check the head when there is no head :)
db0bcb70
amyeroberts
approved these changes
on 2022-08-23
ydshieh
approved these changes
on 2022-08-23
some models w/o head have outputs beyond the hidden layer outputs
c3c6625e
ydshieh
approved these changes
on 2022-08-23
Login to write a write a comment.
Login via GitHub