unofficial-ascend-tools
Project description
Ascend NPU accelerate Embedding model
Please ensure that you haved installed CANN and torch_npu.
Example:
- source the environment
source /usr/local/Ascend/ascend-toolkit/env.sh
-
install torch and torch_npu
-
now use like bellow
from unofficial_ascend_tools.embeddings import AscendEmbeddings
model = AscendEmbeddings(model_path=<path_to_model>,
device_id=0,
query_instruction="Represend this sentence for searching relevant passages: "
)
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
Close
Hashes for unofficial_ascend_tools-0.0.3-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 1b26667a1c25d1106a87f013211d70c33578de5f47eadbc44e9b6da30263822e |
|
| MD5 | 4fd123fc626471fcaea6a6164c52ecf3 |
|
| BLAKE2b-256 | 9bce078ac8f4a51b11f790960a9f6df2ab431a6d58a1e1368ba3dd548d2f8493 |
