search

netease-youdao / BCEmbedding

Netease Youdao's open-source embedding and reranker models for RAG products.
Apache License 2.0
1.46k stars 97 forks source link

Update embedding.py to support Intel Arc Graphics Card #86

Closed Guochengjie closed 1 month ago

Guochengjie commented 2 months ago

It was tested to work on Intel Arc A770 16GB GPU with the latest oneAPI setup.
Intel® oneAPI Base Toolkit: 2024.2.1
Python: 3.11
Windows 11 with driver Intel® Arc™ & Iris® Xe Graphics - WHQL - Windows* 32.0.101.5768

Reference: https://intel.github.io/intel-extension-for-pytorch/index.html#installation?platform=gpu&version=v2.1.40%2bxpu&os=windows&package=pip