Jelajahi Sumber

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 bulan lalu
induk
melakukan
2fa51c19b0
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  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");