Added support for L4&RTX6000 GPU in WebUI (#1738)

* support L4 * add RTX6000
2024-11-30 18:24:32 +01:00 · 2024-01-17 23:56:35 +09:00 · 2024-01-17 23:56:35 +09:00 · 68f286b6a1
commit 68f286b6a1
parent 016e8754b9
1 changed files with 2 additions and 0 deletions
--- a/infer-web.py
+++ b/infer-web.py
@ -93,6 +93,8 @@ if torch.cuda.is_available() or ngpu != 0:
                "M4",
                "T4",
                "TITAN",
+                "L4",
+                "6000",
            ]
        ):
            # A10#A100#V100#A40#P40#M40#K80#A4500