From 68f286b6a1a4e9914d9b2dc8befd14bc8e739428 Mon Sep 17 00:00:00 2001
From: nr1a <82102170+nr1a@users.noreply.github.com>
Date: Wed, 17 Jan 2024 23:56:35 +0900
Subject: [PATCH] Added support for L4&RTX6000 GPU in WebUI (#1738)

* support L4

* add RTX6000
---
 infer-web.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/infer-web.py b/infer-web.py
index 60cd771..6918a92 100644
--- 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