huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Preliminary support for importing PyTorch weights. (#511)

Browse Source 
* Pickle work-in-progress.

* More unpickling.

* More pickling.

* Proper handling of setitems.

* Clippy.

* Again more pickling.

* Restore the example.

* Add enough pickle support to get the list of tensors.

* Read the data from zip files.

* Retrieve the tensor shape.

* Extract the size and dtype.

* More storage types.

* Improve the destructuring.
This commit is contained in:
Laurent Mazare
2023-08-19 11:26:32 +01:00
committed by GitHub
parent 90ff04e77e
commit ad33715c61
3 changed files with 573 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-core/src/lib.rs
View File

@ -56,6 +56,7 @@ pub mod layout;
mod mkl;
pub mod npy;
mod op;
pub mod pickle;
pub mod quantized;
pub mod safetensors;
pub mod shape;