vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix linting

Browse source
This commit is contained in:
Christian Zhou-Zheng 2024-06-09 16:05:08 -04:00
parent 9d7f694438
commit 0417104397
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
gguf-py/gguf/gguf_writer.py
View file

@ -302,11 +302,9 @@ class GGUFWriter:
tensor_shape = quant_shape_from_byte_shape(tensor_shape, raw_dtype) tensor_shape = quant_shape_from_byte_shape(tensor_shape, raw_dtype)
# split when over tensor limit # split when over tensor limit
if (self.split_max_tensors != 0 and \ if (self.split_max_tensors != 0 and len(self.tensors[-1]) >= self.split_max_tensors \
len(self.tensors[-1]) >= self.split_max_tensors \ # or split when over size limit
# 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):
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()) self.tensors.append(dict())