Restored accidentally removed comment

This commit is contained in:
goerch 2023-08-21 00:13:25 +02:00
parent dea1e4c03e
commit 6586487e62

View file

@ -747,6 +747,8 @@ class OutputFile:
self.gguf.add_token_scores(scores) self.gguf.add_token_scores(scores)
self.gguf.add_token_types(toktypes) self.gguf.add_token_types(toktypes)
# TODO: added / special tokens
def add_tensor_info(self, name: str, tensor: LazyTensor) -> None: def add_tensor_info(self, name: str, tensor: LazyTensor) -> None:
n_elements = 1 n_elements = 1
for dim in tensor.shape: for dim in tensor.shape: