py : fix flake8 lint

This commit is contained in:
Georgi Gerganov 2024-01-19 13:52:22 +02:00
parent 7051aacfac
commit de9a147df1

View File

@ -1179,6 +1179,7 @@ class PlamoModel(Model):
self.gguf_writer.add_tensor(new_name, data)
class CodeShellModel(Model):
def set_gguf_parameters(self):
block_count = self.hparams["n_layer"]