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