Skip to content

Commit

Permalink
remove unused deprecated field from MODEL_TABLE
Browse files Browse the repository at this point in the history
  • Loading branch information
@mattf
mattf committed May 30, 2024
1 parent 42048b0 commit c9f8d2b
Showing 1 changed file with 0 additions and 31 deletions.
31 changes: 0 additions & 31 deletions libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,6 @@ class Model(BaseModel):
id="meta/codellama-70b",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=[
"ai-codellama-70b",
"playground_llama2_code_70b",
Expand All @@ -42,14 +41,12 @@ class Model(BaseModel):
id="google/gemma-7b",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-gemma-7b", "playground_gemma_7b", "gemma_7b"],
),
"meta/llama2-70b": Model(
id="meta/llama2-70b",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=[
"ai-llama2-70b",
"playground_llama2_70b",
Expand All @@ -62,126 +59,108 @@ class Model(BaseModel):
id="mistralai/mistral-7b-instruct-v0.2",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-mistral-7b-instruct-v2", "playground_mistral_7b", "mistral_7b"],
),
"mistralai/mixtral-8x7b-instruct-v0.1": Model(
id="mistralai/mixtral-8x7b-instruct-v0.1",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-mixtral-8x7b-instruct", "playground_mixtral_8x7b", "mixtral_8x7b"],
),
"google/codegemma-7b": Model(
id="google/codegemma-7b",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-codegemma-7b"],
),
"google/gemma-2b": Model(
id="google/gemma-2b",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-gemma-2b", "playground_gemma_2b", "gemma_2b"],
),
"google/recurrentgemma-2b": Model(
id="google/recurrentgemma-2b",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-recurrentgemma-2b"],
),
"mistralai/mistral-large": Model(
id="mistralai/mistral-large",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-mistral-large"],
),
"mistralai/mixtral-8x22b-instruct-v0.1": Model(
id="mistralai/mixtral-8x22b-instruct-v0.1",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-mixtral-8x22b-instruct"],
),
"meta/llama3-8b-instruct": Model(
id="meta/llama3-8b-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-llama3-8b-instruct"],
),
"meta/llama3-70b-instruct": Model(
id="meta/llama3-70b-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-llama3-70b-instruct"],
),
"microsoft/phi-3-mini-128k-instruct": Model(
id="microsoft/phi-3-mini-128k-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-phi-3-mini"],
),
"snowflake/arctic": Model(
id="snowflake/arctic",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-arctic"],
),
"databricks/dbrx-instruct": Model(
id="databricks/dbrx-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-dbrx-instruct"],
),
"microsoft/phi-3-mini-4k-instruct": Model(
id="microsoft/phi-3-mini-4k-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-phi-3-mini-4k", "playground_phi2", "phi2"],
),
"seallms/seallm-7b-v2.5": Model(
id="seallms/seallm-7b-v2.5",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-seallm-7b"],
),
"aisingapore/sea-lion-7b-instruct": Model(
id="aisingapore/sea-lion-7b-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-sea-lion-7b-instruct"],
),
"microsoft/phi-3-small-8k-instruct": Model(
id="microsoft/phi-3-small-8k-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-phi-3-small-8k-instruct"],
),
"microsoft/phi-3-small-128k-instruct": Model(
id="microsoft/phi-3-small-128k-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-phi-3-small-128k-instruct"],
),
"microsoft/phi-3-medium-4k-instruct": Model(
id="microsoft/phi-3-medium-4k-instruct",
model_type="chat",
client="ChatNVIDIA",
deprecated=False,
aliases=["ai-phi-3-medium-4k-instruct"],
),
}
Expand All @@ -192,47 +171,41 @@ class Model(BaseModel):
model_type="vlm",
client="ChatNVIDIA",
endpoint="https://ai.api.nvidia.com/v1/vlm/adept/fuyu-8b",
deprecated=False,
aliases=["ai-fuyu-8b", "playground_fuyu_8b", "fuyu_8b"],
),
"google/deplot": Model(
id="google/deplot",
model_type="vlm",
client="ChatNVIDIA",
endpoint="https://ai.api.nvidia.com/v1/vlm/google/deplot",
deprecated=False,
aliases=["ai-google-deplot", "playground_deplot", "deplot"],
),
"microsoft/kosmos-2": Model(
id="microsoft/kosmos-2",
model_type="vlm",
client="ChatNVIDIA",
endpoint="https://ai.api.nvidia.com/v1/vlm/microsoft/kosmos-2",
deprecated=False,
aliases=["ai-microsoft-kosmos-2", "playground_kosmos_2", "kosmos_2"],
),
"nvidia/neva-22b": Model(
id="nvidia/neva-22b",
model_type="vlm",
client="ChatNVIDIA",
endpoint="https://ai.api.nvidia.com/v1/vlm/nvidia/neva-22b",
deprecated=False,
aliases=["ai-neva-22b", "playground_neva_22b", "neva_22b"],
),
"google/paligemma": Model(
id="google/paligemma",
model_type="vlm",
client="ChatNVIDIA",
endpoint="https://ai.api.nvidia.com/v1/vlm/google/paligemma",
deprecated=False,
aliases=["ai-google-paligemma"],
),
"microsoft/phi-3-vision-128k-instruct": Model(
id="microsoft/phi-3-vision-128k-instruct",
model_type="vlm",
client="ChatNVIDIA",
endpoint="https://ai.api.nvidia.com/v1/vlm/microsoft/phi-3-vision-128k-instruct",
deprecated=False,
aliases=["ai-phi-3-vision-128k-instruct"],
),
}
Expand All @@ -242,15 +215,13 @@ class Model(BaseModel):
id="snowflake/arctic-embed-l",
model_type="embedding",
client="NVIDIAEmbeddings",
deprecated=False,
aliases=["ai-arctic-embed-l"],
),
"NV-Embed-QA": Model(
id="NV-Embed-QA",
model_type="embedding",
client="NVIDIAEmbeddings",
endpoint="https://ai.api.nvidia.com/v1/retrieval/nvidia/embeddings",
deprecated=False,
aliases=[
"ai-embed-qa-4",
"playground_nvolveqa_40k",
Expand All @@ -265,7 +236,6 @@ class Model(BaseModel):
model_type="ranking",
client="NVIDIARerank",
endpoint="https://ai.api.nvidia.com/v1/retrieval/nvidia/reranking",
deprecated=False,
aliases=["ai-rerank-qa-mistral-4b"],
),
}
Expand All @@ -275,7 +245,6 @@ class Model(BaseModel):
id="mistralai/mixtral-8x22b-v0.1",
model_type="completion",
client="NVIDIA",
deprecated=False,
aliases=["ai-mixtral-8x22b"],
),
}
Expand Down

0 comments on commit c9f8d2b

Please sign in to comment.