diff --git a/infer-web.py b/infer-web.py index 54a1c4d..488dd7f 100644 --- a/infer-web.py +++ b/infer-web.py @@ -36,6 +36,10 @@ else: or "20" in gpu_name or "30" in gpu_name or "40" in gpu_name + or "A2" in gpu_name.upper() + or "A3" in gpu_name.upper() + or "A4" in gpu_name.upper() + or "P4" in gpu_name.upper() or "A50" in gpu_name.upper() or "70" in gpu_name or "80" in gpu_name