summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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"]