mirror of
https://github.com/huggingface/candle.git
synced 2025-06-19 19:58:35 +00:00
Fix: pth files don't load on Windows (#1661)
* Don't treat zip path as OS path * Add a test case * Add code to generate test pth data
This commit is contained in:
6
candle-core/tests/pth_tests.rs
Normal file
6
candle-core/tests/pth_tests.rs
Normal file
@ -0,0 +1,6 @@
|
||||
/// Regression test for pth files not loading on Windows.
|
||||
#[test]
|
||||
fn test_pth() {
|
||||
let tensors = candle_core::pickle::PthTensors::new("tests/test.pt").unwrap();
|
||||
tensors.get("test").unwrap().unwrap();
|
||||
}
|
Reference in New Issue
Block a user