mirror of
https://github.com/huggingface/candle.git
synced 2025-06-14 18:06:36 +00:00
904bbdae65
* Some first `Module` implementations * Add `state_dict` and `load_state_dict` functionality * Move modules around and create `candle.nn.Linear` * Add `nn.Embedding` and `nn.LayerNorm` * Add BERT implementation * Batch q-matmul * Automatically dequantize `QTensors` if a `Tensor` is expected * Add Module `.to()`, `.cuda()`, `cpu()` and `.type()` functionality * Unittests for `Module`, `Tensor` and `candle.utils` * Add `pytorch` like slicing to `Tensor` * Cleanup and BERT fixes * `black` formatting + unit-test for `nn.Linear` * Refactor slicing implementation
34 lines
673 B
TOML
34 lines
673 B
TOML
[project]
|
|
name = 'candle-nn'
|
|
requires-python = '>=3.7'
|
|
authors = [
|
|
{name = 'The Candle Team'},
|
|
]
|
|
|
|
dynamic = [
|
|
'description',
|
|
'license',
|
|
'readme',
|
|
]
|
|
|
|
[project.urls]
|
|
Homepage = 'https://github.com/huggingface/candle'
|
|
Source = 'https://github.com/huggingface/candle'
|
|
|
|
[build-system]
|
|
requires = ["maturin>=1.0,<2.0"]
|
|
build-backend = "maturin"
|
|
|
|
[tool.maturin]
|
|
python-source = "py_src"
|
|
module-name = "candle.candle"
|
|
bindings = 'pyo3'
|
|
features = ["pyo3/extension-module"]
|
|
|
|
[tool.black]
|
|
line-length = 119
|
|
target-version = ['py35']
|
|
|
|
[project.optional-dependencies]
|
|
testing = ["pytest", "black==22.3"]
|
|
huggingface = ["transformers>=4.33.3", "huggingface-hub>=0.17.3"] |