Support decision transformers - Githubissues

xenova / transformers.js

State-of-the-art Machine Learning for the web. Run 🤗 Transformers directly in your browser, with no need for a server!

https://huggingface.co/docs/transformers.js

Apache License 2.0

9.71k stars 571 forks source link

Support decision transformers #794

Open emilx-cell opened 2 weeks ago

emilx-cell commented 2 weeks ago

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.

xenova commented 2 weeks ago

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.

emilx-cell commented 2 weeks ago

Thank you, looks great! amazing package.