12b2a337f3
Handle start-offset when loading a tensor from a pickle file. ( #1546 )
2024-01-08 09:20:48 +01:00
8a82d623e5
Handle LongStorage in pytorch checkpoints. ( #1152 )
2023-10-22 18:34:36 +01:00
6c588c4792
Refactor the pth tensor exctraction. ( #1109 )
2023-10-16 18:16:34 +01:00
0106b0b04c
Read all the tensors in a PyTorch pth file. ( #1106 )
2023-10-16 13:50:07 +01:00
2cde0cb74b
More pickle support. ( #588 )
...
* More pickle support.
* Be more verbose.
2023-08-24 18:45:10 +01:00
8c232d706b
Small tweaks to the pickle handling to be able to use libtorch files. ( #530 )
...
* Small tweaks to the pickle handling to be able to use libtorch files.
* Move the pytorch specific bits in a different function.
2023-08-20 23:25:34 +01:00
551409092e
Small tweaks to tensor-tools. ( #517 )
2023-08-19 16:50:26 +01:00
6431140250
Retrieve tensor data from PyTorch files. ( #516 )
2023-08-19 15:57:18 +01:00
607ffb9f1e
Retrieve more information from PyTorch checkpoints. ( #515 )
...
* Retrieve more information from PyTorch checkpoints.
* Add enough support to load dino-v2 backbone weights.
2023-08-19 15:05:34 +01:00
ad33715c61
Preliminary support for importing PyTorch weights. ( #511 )
...
* 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.
2023-08-19 11:26:32 +01:00
