diff options
| author | Lukas Kreussel <65088241+LLukas22@users.noreply.github.com> | 2023-10-06 20:01:07 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-06 19:01:07 +0100 |
| commit | 904bbdae65d69aac0c54c29eef744ca5e69c6733 (patch) | |
| tree | 8e191c2cb8cac91d76d2bb9875a60d4ccfe9dbf5 /candle-pyo3/.gitignore | |
| parent | b0442eff8a696d1faba10e23ba645eb11e385116 (diff) | |
| download | candle-904bbdae65d69aac0c54c29eef744ca5e69c6733.tar.gz candle-904bbdae65d69aac0c54c29eef744ca5e69c6733.tar.bz2 candle-904bbdae65d69aac0c54c29eef744ca5e69c6733.zip | |
Make the Python Wrapper more Hackable and simplify Quantization (#1010)
* 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
Diffstat (limited to 'candle-pyo3/.gitignore')
| -rw-r--r-- | candle-pyo3/.gitignore | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-pyo3/.gitignore b/candle-pyo3/.gitignore index 68bc17f9..3d1f96fb 100644 --- a/candle-pyo3/.gitignore +++ b/candle-pyo3/.gitignore @@ -1,3 +1,4 @@ +tests/_workdir # Byte-compiled / optimized / DLL files __pycache__/ *.py[cod] |