diff --git a/infer-web.py b/infer-web.py index f574e71..5d94ee3 100644 --- a/infer-web.py +++ b/infer-web.py @@ -93,8 +93,7 @@ if torch.cuda.is_available() or ngpu != 0: "90", "M4", "T4", - "TITAN", - "L4", + "TITAN","4060", "L", "6000", ] ):