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") ```