Просмотр исходного кода

vocab : correct bounds check for UGM XCDA array access (#17215)

o7si 2 месяцев назад
Родитель
Сommit
ffb6f3d921
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      src/llama-vocab.cpp

+ 1 - 1
src/llama-vocab.cpp

@@ -1013,7 +1013,7 @@ private:
         }
     private:
         uint32_t get_node(size_t index) {
-            if (index > xcda_array_size) {
+            if (index >= xcda_array_size) {
                 throw std::runtime_error("Index out of array bounds in XCDA array!");
             }
             return xcda_array[index];