From caff779dd422624279f3694ac956873663c4f6a0 Mon Sep 17 00:00:00 2001 From: jp Date: Tue, 26 Nov 2024 22:28:58 +0900 Subject: [PATCH] Fix: docs typo (#2777) Fix: typo in model loading code Fix typo in model loading code --- docs/source/supported_models.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/supported_models.md b/docs/source/supported_models.md index 55449e47..0f39ff28 100644 --- a/docs/source/supported_models.md +++ b/docs/source/supported_models.md @@ -43,7 +43,7 @@ If the above list lacks the model you would like to serve, depending on the mode ```python # for causal LMs/text-generation models -AutoModelForCausalLM.from_pretrained(, device_map="auto")` +AutoModelForCausalLM.from_pretrained(, device_map="auto") # or, for text-to-text generation models AutoModelForSeq2SeqLM.from_pretrained(, device_map="auto") ```