transformers.js

transformers.js copied to clipboard

Model description

Hi, I would love to see decision transformers added to the list of supported models: https://huggingface.co/docs/transformers/model_doc/decision_transformer Thank you, EM.

Prerequisites

• [X] The model is supported in Transformers (i.e., listed here)
• [ ] The model can be exported to ONNX with Optimum (i.e., listed here)

Additional information

An example decision model: https://huggingface.co/edbeeching/decision_transformer_atari

Your contribution

I apologize, but this is beyond my coding skills.

Here we go: https://github.com/xenova/transformers.js/pull/795. + list of converted models: https://huggingface.co/models?library=transformers.js&other=decision_transformer&sort=trending

I will add example Transformers.js code now too. I'll also look at the model you linked.

Thank you, looks great! amazing package.