Add A100e GPU type (#299)

yunfeng-scale · web-flow · commit c47751add472 · 2023-09-29T12:38:52.000-07:00
diff --git a/clients/python/llmengine/data_types.py b/clients/python/llmengine/data_types.py
@@ -32,6 +32,7 @@ class GpuType(str, Enum):
     NVIDIA_TESLA_T4 = "nvidia-tesla-t4"
     NVIDIA_AMPERE_A10 = "nvidia-ampere-a10"
     NVIDIA_AMPERE_A100 = "nvidia-ampere-a100"
+    NVIDIA_AMPERE_A100E = "nvidia-ampere-a100e"
 
 
 class ModelEndpointType(str, Enum):
