Skip to content

GitLab

Unverified Commit 1d2cb356 authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Fix doc. (#2792)

parent d4718051
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
update_doc.py
View file @ 1d2cb356
...@@ -16,7 +16,7 @@ If the above list lacks the model you would like to serve, depending on the mode ...@@ -16,7 +16,7 @@ If the above list lacks the model you would like to serve, depending on the mode
```python ```python
# for causal LMs/text-generation models # for causal LMs/text-generation models
AutoModelForCausalLM.from_pretrained(<model>, device_map="auto")` AutoModelForCausalLM.from_pretrained(<model>, device_map="auto")
# or, for text-to-text generation models # or, for text-to-text generation models
AutoModelForSeq2SeqLM.from_pretrained(<model>, device_map="auto") AutoModelForSeq2SeqLM.from_pretrained(<model>, device_map="auto")
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment