huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: typo in model loading code

Browse Source 
Fix typo in model loading code
This commit is contained in:
jp 2024-11-25 10:25:32 +09:00 committed by GitHub
parent 780531ec77
commit a1dd3ffe25
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/supported_models.md
View File

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