summaryrefslogtreecommitdiff
path: root/convert-hf-to-gguf.py
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2024-01-19 13:52:22 +0200
committerGeorgi Gerganov <ggerganov@gmail.com>2024-01-19 13:52:22 +0200
commitde9a147df14e62f54f879d2d15e6c4793107f4fc (patch)
tree6d4d7810b4a008462f3d0206796df6ab6f9c5d9c /convert-hf-to-gguf.py
parent7051aacfac0057fa5fac9ea46c55bffc3892d810 (diff)
py : fix flake8 lint
Diffstat (limited to 'convert-hf-to-gguf.py')
-rwxr-xr-xconvert-hf-to-gguf.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
index aae3a5e8..d2d6948d 100755
--- a/convert-hf-to-gguf.py
+++ b/convert-hf-to-gguf.py
@@ -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"]