Commit 64b3da6
Update pipelines.js to allow for
* Update pipelines.js to allow for `token_embeddings` as well
In recent examples of optimum pipeline export the feature extraction pipelines have their output state as `token_embeddings` instead of `last_hidden_state`.
So we should support this as well.
* Keep diff small
* Keep diff small
---------
Co-authored-by: Joshua Lochner <[email protected]>token_embeddings as well (#770)1 parent 250fcd7 commit 64b3da6
1 file changed
+1
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1255 | 1255 | | |
1256 | 1256 | | |
1257 | 1257 | | |
1258 | | - | |
| 1258 | + | |
1259 | 1259 | | |
1260 | 1260 | | |
1261 | 1261 | | |
| |||
0 commit comments