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: "
)
Project details
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
File details
Details for the file unofficial_ascend_tools-0.0.3-py3-none-any.whl
.
File metadata
- Download URL: unofficial_ascend_tools-0.0.3-py3-none-any.whl
- Upload date:
- Size: 7.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.10.14
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1b26667a1c25d1106a87f013211d70c33578de5f47eadbc44e9b6da30263822e |
|
MD5 | 4fd123fc626471fcaea6a6164c52ecf3 |
|
BLAKE2b-256 | 9bce078ac8f4a51b11f790960a9f6df2ab431a6d58a1e1368ba3dd548d2f8493 |