Update CLI Gpu types (#167)

sam-scale · web-flow · commit 43adc122cf06 · 2024-06-11T09:58:47.000-04:00
* Update CLI Gpu types

* Update endpoints.py
diff --git a/launch/cli/endpoints.py b/launch/cli/endpoints.py
@@ -205,7 +205,7 @@ def edit_endpoint(ctx: click.Context, endpoint_name: str):
 gpu_type_prompt = "Gpu type (optional): " if gpus == 0 else "Gpu type: "
 gpu_type = q.select(
 gpu_type_prompt,
- choices=["", "nvidia-tesla-t4", "nvidia-ampere-a10", "nvidia-a100"],
+ choices=["", "nvidia-hopper-h100-1g20g", "nvidia-hopper-h100-3g40g", "nvidia-hopper-h100", "nvidia-tesla-t4", "nvidia-ampere-a10", "nvidia-ampere-a100"],
 ).ask()
 min_workers = q.text(
 "Min workers: ",

-Original file line number
+Diff line change
 gpu_type_prompt="Gpu type (optional): "ifgpus==0else"Gpu type: "
 gpu_type=q.select(
 gpu_type_prompt,
 -choices=["", "nvidia-tesla-t4", "nvidia-ampere-a10", "nvidia-a100"],
 +choices=["", "nvidia-hopper-h100-1g20g", "nvidia-hopper-h100-3g40g", "nvidia-hopper-h100", "nvidia-tesla-t4", "nvidia-ampere-a10", "nvidia-ampere-a100"],
  ).ask()
 min_workers=q.text(
 "Min workers: ",