Update app.py
Browse files
app.py
CHANGED
|
@@ -14,6 +14,8 @@ from transformers import AutoModelForCausalLM, GPT2LMHeadModel, PreTrainedTokeni
|
|
| 14 |
|
| 15 |
import chess
|
| 16 |
|
|
|
|
|
|
|
| 17 |
class UciTokenizer(PreTrainedTokenizerFast):
|
| 18 |
_PAD_TOKEN: str
|
| 19 |
_UNK_TOKEN: str
|
|
@@ -299,7 +301,6 @@ def setup_app(model: GPT2LMHeadModel):
|
|
| 299 |
|
| 300 |
return iface
|
| 301 |
|
| 302 |
-
checkpoint_name = "austindavis/gpt2-lichess-uci-202306"
|
| 303 |
model: GPT2LMHeadModel = AutoModelForCausalLM.from_pretrained(checkpoint_name)
|
| 304 |
model.requires_grad_(False)
|
| 305 |
|
|
|
|
| 14 |
|
| 15 |
import chess
|
| 16 |
|
| 17 |
+
checkpoint_name = "austindavis/chess-gpt2-uci-8x8x512"
|
| 18 |
+
|
| 19 |
class UciTokenizer(PreTrainedTokenizerFast):
|
| 20 |
_PAD_TOKEN: str
|
| 21 |
_UNK_TOKEN: str
|
|
|
|
| 301 |
|
| 302 |
return iface
|
| 303 |
|
|
|
|
| 304 |
model: GPT2LMHeadModel = AutoModelForCausalLM.from_pretrained(checkpoint_name)
|
| 305 |
model.requires_grad_(False)
|
| 306 |
|