Spaces:
Running
Running
update tokenizer to use 36 character version
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ device = 'cpu'
|
|
25 |
model_id = "philipp-zettl/chessPT"
|
26 |
|
27 |
model_path = hf_hub_download(repo_id=model_id, filename="chessPT-v0.5.pth")
|
28 |
-
tokenizer_path = hf_hub_download(repo_id=model_id, filename="tokenizer.json")
|
29 |
|
30 |
model = DecoderTransformer(vocab_size, n_embed, context_size, n_layer, n_head, dropout)
|
31 |
model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
|
|
|
25 |
model_id = "philipp-zettl/chessPT"
|
26 |
|
27 |
model_path = hf_hub_download(repo_id=model_id, filename="chessPT-v0.5.pth")
|
28 |
+
tokenizer_path = hf_hub_download(repo_id=model_id, filename="tokenizer-v0.5.json")
|
29 |
|
30 |
model = DecoderTransformer(vocab_size, n_embed, context_size, n_layer, n_head, dropout)
|
31 |
model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
|