speakrs 0.6.0

Fast Rust speaker diarization with pyannote-level accuracy and native CPU/CoreML/CUDA inference
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

//! In-memory weights and tensor inspection for CUDA tests

use super::{Path, SafetensorsFile};

impl SafetensorsFile {
    /// The file this was read from
    pub fn path(&self) -> &Path {
        self.0.path()
    }

    /// The stored shape of a tensor, if the file has it
    pub fn shape(&self, name: &str) -> Option<&[usize]> {
        self.0.shape(name)
    }
}

mod tests {
    use serde_json::json;

    use crate::inference::native_model::test_support::TestFile;

    use super::super::super::CudaError;
    use super::SafetensorsFile;

    #[test]
    fn adapter_keeps_cuda_weight_error_variants_and_payloads() {
        let file = TestFile::new(
            &json!({
                "float": {"dtype": "F32", "shape": [1], "data_offsets": [0, 4]},
                "other": {"dtype": "U8", "shape": [1], "data_offsets": [4, 5]}
            }),
            &[0, 0, 128, 63, 7],
        );
        let weights = SafetensorsFile::open(&file.0).unwrap();
        assert_eq!(weights.host().names(), ["float", "other"]);
        assert_eq!(weights.read_f32("float", &[1]).unwrap(), [1.0]);
        assert!(matches!(weights.read_f32("missing", &[1]),
            Err(CudaError::MissingTensor { path, name })
                if path == file.0 && name == "missing"));
        assert!(matches!(weights.read_f32("other", &[1]),
            Err(CudaError::TensorDtype { name, dtype })
                if name == "other" && dtype == "U8"));
        assert!(matches!(weights.read_f32("float", &[2]),
            Err(CudaError::TensorShape { name, expected, actual })
                if name == "float" && expected == [2] && actual == [1]));

        std::fs::write(&file.0, [0, 1, 2]).unwrap();
        assert!(matches!(SafetensorsFile::open(&file.0),
            Err(CudaError::WeightsFormat { path, .. }) if path == file.0));
        let absent = file.0.with_extension("absent");
        assert!(matches!(SafetensorsFile::open(&absent),
            Err(CudaError::WeightsIo { path, source })
                if path == absent && source.kind() == std::io::ErrorKind::NotFound));
    }
}