in bindings/python/src/lib.rs [313:322]
fn parse_device(name: &str) -> PyResult<usize> {
let tokens: Vec<_> = name.split(':').collect();
if tokens.len() == 2 {
Ok(tokens[1].parse()?)
} else {
Err(SafetensorError::new_err(format!(
"device {name} is invalid"
)))
}
}