mirror of
https://github.com/huggingface/candle.git
synced 2025-06-20 04:00:28 +00:00
Pickle support: dig within the _rebuild_parameter calls. (#1681)
This commit is contained in:
@ -217,6 +217,13 @@ impl Object {
|
|||||||
let args = args.remove(1);
|
let args = args.remove(1);
|
||||||
(callable, args)
|
(callable, args)
|
||||||
}
|
}
|
||||||
|
Object::Class {
|
||||||
|
module_name,
|
||||||
|
class_name,
|
||||||
|
} if module_name == "torch._utils" && class_name == "_rebuild_parameter" => {
|
||||||
|
let mut args = args.tuple()?;
|
||||||
|
args.remove(0).reduce()?
|
||||||
|
}
|
||||||
_ => (callable, args),
|
_ => (callable, args),
|
||||||
};
|
};
|
||||||
match callable {
|
match callable {
|
||||||
|
Reference in New Issue
Block a user