huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support defaultdict in PyTorch checkpoints. (#1696)

Browse Source 
* Support defaultdict in PyTorch checkpoints.

* Fix clippy lint.
This commit is contained in:
Laurent Mazare
2024-02-12 10:26:56 +01:00
committed by GitHub
parent 1e26d539d9
commit 274bf11633
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/src/pickle.rs
View File

@ -350,8 +350,10 @@ impl Stack {
module_name,
class_name,
} => {
if module_name == "collections" && class_name == "OrderedDict" {
// TODO: have a separate ordered dict.
if module_name == "collections"
&& (class_name == "OrderedDict" || class_name == "defaultdict")
{
// TODO: have a separate ordered dict and a separate default dict.
Some(Object::Dict(vec![]))
} else {
None