diff --git a/R/safetensors.R b/R/safetensors.R index c1c7706..603b2be 100644 --- a/R/safetensors.R +++ b/R/safetensors.R @@ -158,7 +158,9 @@ torch_dtype_from_safe <- function(x) { "I8" = "int8", "I16" = "int16", "I32" = "int32", - "I64" = "int64" + "I64" = "int64", + "BF16" = "bfloat16", + cli::cli_abort("Unsupported dtype {.val {x}}") ) }