From a1dd3ffe259a5350683b88192d79459ee83fa3ab Mon Sep 17 00:00:00 2001 From: jp Date: Mon, 25 Nov 2024 10:25:32 +0900 Subject: [PATCH] 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") ```