From: Justine Tunney <redacted>
Date: Wed, 22 May 2024 11:08:18 +0000 (-0400)
Subject: llama : add missing model type names (#7445)
X-Git-Tag: upstream/0.0.4488~1522
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=03d8900ebe062355e26a562379daee5f17ea099f;p=pkg%2Fggml%2Fsources%2Fllama.cpp

llama : add missing model type names (#7445)
---

diff --git a/llama.cpp b/llama.cpp
index abff8c1c..d8c6f29a 100644
--- a/llama.cpp
+++ b/llama.cpp
@@ -3771,14 +3771,17 @@ static std::string llama_model_ftype_name(llama_ftype ftype) {
 
 static const char * llama_model_type_name(e_model type) {
     switch (type) {
+        case MODEL_17M:    return "17M";
         case MODEL_22M:    return "22M";
         case MODEL_33M:    return "33M";
         case MODEL_109M:   return "109M";
         case MODEL_137M:   return "137M";
+        case MODEL_335M:   return "335M";
         case MODEL_0_5B:   return "0.5B";
         case MODEL_1B:     return "1B";
         case MODEL_2B:     return "2B";
         case MODEL_3B:     return "3B";
+        case MODEL_4B:     return "4B";
         case MODEL_7B:     return "7B";
         case MODEL_8B:     return "8B";
         case MODEL_12B:    return "12B";