AppleSwing
commited on
Commit
•
0fb715c
1
Parent(s):
dbe8db4
Fix a bug
Browse files- backend-cli.py +1 -1
backend-cli.py
CHANGED
@@ -449,7 +449,7 @@ def get_args():
|
|
449 |
parser.add_argument("--inference-framework", type=str, default="hf-chat", help="Inference framework to debug")
|
450 |
parser.add_argument("--limit", type=int, default=None, help="Limit for the number of samples")
|
451 |
parser.add_argument("--gpu-type", type=str, default="NVIDIA-A100-PCIe-80GB",
|
452 |
-
help="GPU type. NVIDIA-A100-PCIe-80GB; NVIDIA-RTX-A5000-
|
453 |
return parser.parse_args()
|
454 |
|
455 |
|
|
|
449 |
parser.add_argument("--inference-framework", type=str, default="hf-chat", help="Inference framework to debug")
|
450 |
parser.add_argument("--limit", type=int, default=None, help="Limit for the number of samples")
|
451 |
parser.add_argument("--gpu-type", type=str, default="NVIDIA-A100-PCIe-80GB",
|
452 |
+
help="GPU type. NVIDIA-A100-PCIe-80GB; NVIDIA-RTX-A5000-24GB; NVIDIA-H100-PCIe-80GB")
|
453 |
return parser.parse_args()
|
454 |
|
455 |
|