Handle LongStorage in pytorch checkpoints. (#1152)

2025-06-18 19:47:12 +00:00 · 2023-10-22 18:34:36 +01:00
parent df2f89b6cf
commit 8a82d623e5
2 changed files with 21 additions and 27 deletions
--- a/candle-core/src/pickle.rs
+++ b/candle-core/src/pickle.rs
@ -609,6 +609,7 @@ fn rebuild_args(args: Object) -> Result<(Layout, DType, String, usize)> {
        "HalfStorage" => DType::F16,
        "BFloat16Storage" => DType::BF16,
        "ByteStorage" => DType::U8,
+        "LongStorage" => DType::I64,
        other => {
            crate::bail!("unsupported storage type {other}")
        }