From 0417104397e54dce71cd2c6e9c23b11f2acf0d60 Mon Sep 17 00:00:00 2001 From: Christian Zhou-Zheng Date: Sun, 9 Jun 2024 16:05:08 -0400 Subject: [PATCH] fix linting --- gguf-py/gguf/gguf_writer.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index 19556c552..69a4a3ff6 100644 --- a/gguf-py/gguf/gguf_writer.py +++ b/gguf-py/gguf/gguf_writer.py @@ -302,11 +302,9 @@ class GGUFWriter: tensor_shape = quant_shape_from_byte_shape(tensor_shape, raw_dtype) # split when over tensor limit - if (self.split_max_tensors != 0 and \ - len(self.tensors[-1]) >= self.split_max_tensors \ - # or split when over size limit - or self.split_max_size != 0 and \ - sum(ti.nbytes for ti in self.tensors[-1].values()) + tensor_nbytes > self.split_max_size): + if (self.split_max_tensors != 0 and len(self.tensors[-1]) >= self.split_max_tensors \ + # or split when over size limit + or self.split_max_size != 0 and sum(ti.nbytes for ti in self.tensors[-1].values()) + tensor_nbytes > self.split_max_size): self.tensors.append(dict())