transformers.js/scripts
Dave 5bb8d25337
Add StableLM support (#616)
* add stablelm model impl.

* add stablelm mapping

* Update `StableLMPreTrainedModel` class config

* Update stablelm conversion script quantization settings

* Add StableLm to list of supported models

* Remove StableLMForSequenceClassification

will be added in seperate PR

Co-authored-by: Joshua Lochner <admin@xenova.com>

* Remove SequenceClassifierOutputWithPast

will be added in seperate PR

Co-authored-by: Joshua Lochner <admin@xenova.com>

* Remove StableLMForSequenceClassification mapping

will be added in seperate pr

Co-authored-by: Joshua Lochner <admin@xenova.com>

* Update src/models.js

* Naming nit

* Naming nits

* Update scripts/convert.py

* Update src/models.js

---------

Co-authored-by: Joshua Lochner <admin@xenova.com>
2024-03-07 03:05:27 +02:00
..
extra Add support for cross-encoder models (+fix token type ids) (#501) 2024-01-04 18:37:26 +02:00
convert.py Add StableLM support (#616) 2024-03-07 03:05:27 +02:00
requirements.txt Use `transformers==4.33.2` in conversion script 2023-09-23 17:06:27 +02:00
supported_models.py Add StableLM support (#616) 2024-03-07 03:05:27 +02:00