EngineX-Ascend/enginex-ascend-910-fe
10
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

add support for a100

Browse Source
This commit is contained in:
ZHANG Hao
2025-09-03 11:01:20 +08:00
parent f0425d03b4
commit fcf2001f24
3 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test.sh Executable file
View File

@@ -0,0 +1,2 @@
# python main.py dataset.json output.json /mnt/contest_ceph/zhanghao/models/BAAI/bge-large-zh-v1.5 --device npu
python main.py dataset.json output.json /mnt/contest_ceph/zhanghao/models/BAAI/bge-large-zh-v1.5 --device cuda