Ver código fonte

model-conversion : add token ids to prompt token output [no ci] (#17863)

This commit adds the token ids to the printed prompt outputs.

The motivation for this is that is can be useful to see the actual token
ids alongside the token strings for debugging.
Daniel Bevenius 1 mês atrás
pai
commit
2fa51c19b0
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      examples/model-conversion/logits.cpp

+ 1 - 1
examples/model-conversion/logits.cpp

@@ -144,7 +144,7 @@ int main(int argc, char ** argv) {
             return 1;
         }
         std::string s(buf, n);
-        printf("%s", s.c_str());
+        printf("%s (%d)", s.c_str(), id);
     }
     printf("\n");