huggingface / optimum-neuron

Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.

Apache License 2.0

176 stars 51 forks source link

Add support for Mixtral #569

Closed dacorvo closed 2 months ago

dacorvo commented 2 months ago

What does this PR do?

Beta support for Mixtral has been added to AWS Neuron SDK 2.18. This enables this model type for NeuronModelForCausalLM and adds the corresponding unit tests.

HuggingFaceDocBuilderDev commented 2 months ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.