Skip to content

Commit 5cde238

Browse files
Aryanag2mrDzurb
andauthored
[AQUA] Extend the CONTAINER_FAMILY_COMPATIBILITY to include GPT-OSS models. (#1253)
Co-authored-by: Dmitrii Cherkasov <[email protected]>
1 parent f8b2f5f commit 5cde238

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

ads/aqua/common/enums.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,12 @@ class Platform(ExtendedEnum):
123123
# - Key: The preferred container family to use when multiple compatible families are selected.
124124
# - Value: A list of all compatible families (including the preferred one).
125125
CONTAINER_FAMILY_COMPATIBILITY: Dict[str, List[str]] = {
126+
InferenceContainerTypeFamily.AQUA_VLLM_OPENAI_CONTAINER_FAMILY: [
127+
InferenceContainerTypeFamily.AQUA_VLLM_OPENAI_CONTAINER_FAMILY,
128+
InferenceContainerTypeFamily.AQUA_VLLM_LLAMA4_CONTAINER_FAMILY,
129+
InferenceContainerTypeFamily.AQUA_VLLM_V1_CONTAINER_FAMILY,
130+
InferenceContainerTypeFamily.AQUA_VLLM_CONTAINER_FAMILY,
131+
],
126132
InferenceContainerTypeFamily.AQUA_VLLM_V1_CONTAINER_FAMILY: [
127133
InferenceContainerTypeFamily.AQUA_VLLM_V1_CONTAINER_FAMILY,
128134
InferenceContainerTypeFamily.AQUA_VLLM_CONTAINER_FAMILY,

0 commit comments

Comments
 (0)