From 1d2cb356b93244959371202ff2c770cfb325f3da Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Mon, 2 Dec 2024 09:58:26 +0530 Subject: [PATCH] Fix doc. (#2792) --- update_doc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/update_doc.py b/update_doc.py index 6357cc00..28430b1a 100644 --- a/update_doc.py +++ b/update_doc.py @@ -16,7 +16,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") ```