Skip to content

Commit 343aa6f

Browse files
authored
Update infer-web.py
1 parent 4ce1528 commit 343aa6f

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

infer-web.py

+4
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,10 @@
3636
or "20" in gpu_name
3737
or "30" in gpu_name
3838
or "40" in gpu_name
39+
or "A2" in gpu_name.upper()
40+
or "A3" in gpu_name.upper()
41+
or "A4" in gpu_name.upper()
42+
or "P4" in gpu_name.upper()
3943
or "A50" in gpu_name.upper()
4044
or "70" in gpu_name
4145
or "80" in gpu_name

0 commit comments

Comments
 (0)