ソースを参照

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 ヶ月 前
コミット
2fa51c19b0
1 ファイル変更1 行追加1 行削除
  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");