kj7rrv/gptc
1
0
Fork 0
Code Issues 2 Pull Requests Projects Releases 12 Wiki Activity

Add type annotation to Model.get()

Browse Source
This commit is contained in:
Samuel Sloniker 2022-11-27 13:36:49 -08:00
parent e4eb322aa7
commit 8d42a92848
Signed by: kj7rrv
GPG Key ID: 1BB4029E66285A62
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gptc/model.py
View File

@ -62,7 +62,7 @@ class Model:
} }
return probs return probs
def get(self, token): def get(self, token: str) -> Dict[str, float]:
try: try:
weights = self.weights[ weights = self.weights[
gptc.tokenizer.hash_single(gptc.tokenizer.normalize(token)) gptc.tokenizer.hash_single(gptc.tokenizer.normalize(token))