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/test.py | |
| 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/test.py')
| -rw-r--r-- | candle-pyo3/test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-pyo3/test.py b/candle-pyo3/test.py index 7f24b49d..a56ed22c 100644 --- a/candle-pyo3/test.py +++ b/candle-pyo3/test.py @@ -7,7 +7,7 @@ print(t + t) t = candle.Tensor([3.0, 1, 4, 1, 5, 9, 2, 6]) print(t) -print(t+t) +print(t + t) t = t.reshape([2, 4]) print(t.matmul(t.t())) |