From 01a0f206dfaf47f22f40b3a231563b0677686e1f Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Sat, 27 May 2023 13:35:40 +0800
Subject: [PATCH] added support for starcoder, which is basically gpt2

---
 model_adapter.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model_adapter.cpp b/model_adapter.cpp
index 5612da902..6864e0d37 100644
--- a/model_adapter.cpp
+++ b/model_adapter.cpp
@@ -126,7 +126,7 @@ void print_tok_vec(std::vector<float> &embd)
                fileformat = FileFormat::GPTJ_3; //quantized format cannot be legacy type
            }
        }
-       else if(vocabsiz==50257)
+       else if(vocabsiz==50257 || (vocabsiz>=49152&&vocabsiz<=49157)) //49152-6 is starcoder
        {
            fileformat = FileFormat::GPT2_1;
            uint32_t temp;