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