Skip to content

Commit c908015

Browse files
committed
Handle empty tensors
1 parent d90034b commit c908015

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/tensors/treetransformers.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ function _transformer_weight((matrix, structures_dst, structures_src))
103103
end
104104

105105
function buffersize(transformer::GenericTreeTransformer)
106-
return maximum(transformer.data) do (basistransform, structures_dst, _)
106+
return maximum(transformer.data; init=0) do (basistransform, structures_dst, _)
107107
return prod(structures_dst[1][1]) * max(size(basistransform)...)
108108
end
109109
end

0 commit comments

Comments
 (0)