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

Minor doc fixes

Browse Source
This commit is contained in:
Vaibhav Srivastav 2024-08-09 12:29:20 +02:00
parent 6d06473cf4
commit 2bd9129f11
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/text_generation_server/models/__init__.py
View File

@ -180,7 +180,7 @@ class ModelType(enum.Enum):
LLAMA = { LLAMA = {
"type": "llama", "type": "llama",
"name": "Llama", "name": "Llama",
"url": "https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct", "url": "https://huggingface.co/collections/meta-llama/llama-31-669fc079a0c406a149a5738f",
} }
PHI3 = { PHI3 = {
"type": "phi3", "type": "phi3",
@ -200,7 +200,7 @@ class ModelType(enum.Enum):
GEMMA2 = { GEMMA2 = {
"type": "gemma2", "type": "gemma2",
"name": "Gemma2", "name": "Gemma2",
"url": "https://huggingface.co/google/gemma2-9b", "url": "https://huggingface.co/collections/google/gemma-2-release-667d6600fd5220e7b967f315",
} }
COHERE = { COHERE = {
"type": "cohere", "type": "cohere",
@ -220,7 +220,7 @@ class ModelType(enum.Enum):
MISTRAL = { MISTRAL = {
"type": "mistral", "type": "mistral",
"name": "Mistral", "name": "Mistral",
"url": "https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2", "url": "https://huggingface.co/mistralai/Mistral-Nemo-Instruct-2407",
} }
MIXTRAL = { MIXTRAL = {
"type": "mixtral", "type": "mixtral",

2
update_doc.py
View File

@ -7,7 +7,7 @@ import os
TEMPLATE = """ TEMPLATE = """
# Supported Models and Hardware # Supported Models and Hardware
Text Generation Inference enables serving optimized models on specific hardware for the highest performance. The following sections list which models are hardware are supported. Text Generation Inference enables serving optimized models on specific hardware for the highest performance. The following sections list which models (VLMs & LLMs) are supported.
## Supported Models ## Supported Models