From b9c664ab2f665a70f53f29cc82753a37418de76d Mon Sep 17 00:00:00 2001
From: Galunid <karolek1231456@gmail.com>
Date: Tue, 31 Oct 2023 03:42:55 +0100
Subject: [PATCH] Woops

---
 model.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/model.py b/model.py
index a15682a34..814623f23 100644
--- a/model.py
+++ b/model.py
@@ -53,7 +53,8 @@ class Model:
             return gguf.MODEL_ARCH.STARCODER
         if arch == "GPTRefactForCausalLM":
             return gguf.MODEL_ARCH.REFACT
-        if arch == ""
+        if arch == "PersimmonForCausalLM":
+            return gguf.MODEL_ARCH.PERSIMMON
 
         raise NotImplementedError(f'Architecture "{arch}" not supported!')
 
@@ -194,6 +195,8 @@ class Model:
             return StarCoderModel
         if model_architecture == "GPTRefactForCausalLM":
             return RefactModel
+        if model_architecture == "PersimmonForCausalLM":
+            return PersimmonModel
         return Model
 
 class StableLMModel(Model):