From 71f460578d5c85739caf42dc6f52e1c2eb12492d Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 22 Dec 2023 10:40:22 +0100 Subject: [PATCH] Update `docs/source/en/perf_infer_gpu_one.md` (#28198) fix Co-authored-by: ydshieh --- docs/source/en/perf_infer_gpu_one.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docs/source/en/perf_infer_gpu_one.md b/docs/source/en/perf_infer_gpu_one.md index 4f25c9fa6f..5cc9cd208d 100644 --- a/docs/source/en/perf_infer_gpu_one.md +++ b/docs/source/en/perf_infer_gpu_one.md @@ -172,6 +172,8 @@ For now, Transformers supports SDPA inference and training for the following arc * [Llama](https://huggingface.co/docs/transformers/model_doc/llama#transformers.LlamaModel) * [Idefics](https://huggingface.co/docs/transformers/model_doc/idefics#transformers.IdeficsModel) * [Whisper](https://huggingface.co/docs/transformers/model_doc/whisper#transformers.WhisperModel) +* [Mistral](https://huggingface.co/docs/transformers/model_doc/mistral#transformers.MistralModel) +* [Mixtral](https://huggingface.co/docs/transformers/model_doc/mixtral#transformers.MixtralModel)