Update webui.py

NVIDIA RTX 6000 Ada Generation
7 months ago · a1fe2267af
parent 793b0043de
commit a1fe2267af
1 changed files with 1 additions and 1 deletions
--- a/webui.py
+++ b/webui.py
@ -72,7 +72,7 @@ mem = []
 if_gpu_ok = False

 # 判断是否有能用来训练和加速推理的N卡
-ok_gpu_keywords={"10","16","20","30","40","A2","A3","A4","P4","A50","500","A60","70","80","90","M4","T4","TITAN","L4","4060","H"}
+ok_gpu_keywords={"10","16","20","30","40","A2","A3","A4","P4","A50","500","A60","70","80","90","M4","T4","TITAN","L4","4060","H","600"}
 set_gpu_numbers=set()
 if torch.cuda.is_available() or ngpu != 0:
    for i in range(ngpu):