Show author and SP length in soft prompt menu

2025-06-05 21:59:24 +02:00 · 2022-01-18 16:30:09 -05:00
parent 1e2dbd42a6
commit 1951ccd2ce
5 changed files with 15 additions and 9 deletions
--- a/fileops.py
+++ b/fileops.py
@ -161,6 +161,7 @@ def getspfiles(model_dimension: int):
            ob = {}
        z.close()
        ob["filename"] = file
+        ob["n_tokens"] = shape[-2]
        lst.append(ob)
    return lst