Skip to content

GitLab

Unverified Commit caff779d authored by jp's avatar jp Committed by GitHub
Browse files

Fix: docs typo (#2777) 

Fix: typo in model loading code

Fix typo in model loading code
parent 892a26e5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/supported_models.md
View file @ caff779d
...@@ -43,7 +43,7 @@ If the above list lacks the model you would like to serve, depending on the mode ...@@ -43,7 +43,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