diff --git a/infer-web.py b/infer-web.py index 8c5f021..6eb795c 100644 --- a/infer-web.py +++ b/infer-web.py @@ -92,7 +92,8 @@ if torch.cuda.is_available() or ngpu != 0: "90", "M4", "T4", - "TITAN", + "TITAN","4060", "L", + "6000", ] ): # A10#A100#V100#A40#P40#M40#K80#A4500