Jelajahi Sumber

Do not crash when it has nothing to say. (#796)

Otherwise observing this in the interactive mode:
/usr/lib/gcc/x86_64-pc-linux-gnu/12/include/g++-v12/bits/stl_vector.h:1230: reference std::vector<int>::back() [_Tp = int, _Alloc = std::allocator<int>]: Assertion '!this->empty()' failed.
Sergey Alirzaev 2 tahun lalu
induk
melakukan
cc9cee8e9e
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      examples/main/main.cpp

+ 1 - 1
examples/main/main.cpp

@@ -431,7 +431,7 @@ int main(int argc, char ** argv) {
         }
 
         // end of text token
-        if (embd.back() == llama_token_eos()) {
+        if (!embd.empty() && embd.back() == llama_token_eos()) {
             if (params.instruct) {
                 is_interacting = true;
             } else {