Update infer-web.py

2023-04-28 11:31:13 +08:00 · 2023-04-28 11:31:13 +08:00 · 211a842e88
parent 9068d5283e
commit 211a842e88
1 changed files with 1542 additions and 1541 deletions
--- a/infer-web.py
+++ b/infer-web.py
@ -34,6 +34,7 @@ else:
        gpu_name = torch.cuda.get_device_name(i)
        if (
            "10" in gpu_name
+            or "16" in gpu_name
            or "20" in gpu_name
            or "30" in gpu_name
            or "40" in gpu_name