huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

PyO3 is back.

Browse Source
This commit is contained in:
Nicolas Patry
2023-07-25 22:44:30 +02:00
parent ed58de7551
commit b7814f66b4
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@ -3,7 +3,7 @@ members = [
"candle-core",
"candle-examples",
"candle-nn",
# "candle-pyo3",
"candle-pyo3",
"candle-transformers",
"candle-wasm-examples/llama2-c",
"candle-wasm-examples/whisper",

3
candle-pyo3/src/lib.rs
View File

@ -14,6 +14,9 @@ pub fn wrap_err(err: ::candle::Error) -> PyErr {
#[pyclass(name = "Tensor")]
struct PyTensor(Tensor);
unsafe impl Send for PyTensor {}
unsafe impl Sync for PyTensor {}
impl std::ops::Deref for PyTensor {
type Target = Tensor;