mirror of
https://github.com/ml-explore/mlx.git
synced 2025-11-02 09:18:11 +08:00
format
This commit is contained in:
@@ -58,7 +58,8 @@ Shape get_shape(const gguf_tensor& tensor) {
|
|||||||
|
|
||||||
std::tuple<allocator::Buffer, Dtype> extract_tensor_data(gguf_tensor* tensor) {
|
std::tuple<allocator::Buffer, Dtype> extract_tensor_data(gguf_tensor* tensor) {
|
||||||
if (tensor == nullptr) {
|
if (tensor == nullptr) {
|
||||||
throw std::invalid_argument("[extract_tensor_data] Input tensor pointer is null.");
|
throw std::invalid_argument(
|
||||||
|
"[extract_tensor_data] Input tensor pointer is null.");
|
||||||
}
|
}
|
||||||
std::optional<Dtype> equivalent_dtype = gguf_type_to_dtype(tensor->type);
|
std::optional<Dtype> equivalent_dtype = gguf_type_to_dtype(tensor->type);
|
||||||
// If there's an equivalent type, we can simply copy.
|
// If there's an equivalent type, we can simply copy.
|
||||||
|
|||||||
Reference in New Issue
Block a user