kaupane
/

ChessFormer-SL

Reinforcement Learning

Model card Files Files and versions

kaupane commited on Jun 28

Commit

9ac07b6

·

verified ·

1 Parent(s): 4b96c5e

Update model.py

Files changed (1) hide show

model.py +2 -2

model.py CHANGED Viewed

@@ -5,9 +5,9 @@ from typing import List, Dict, Tuple
 from huggingface_hub import PyTorchModelHubMixin
 try:
-    from mappings import MAX_HALFMOVES, MAX_FULLMOVES, EMPTY_SQ_IDX, PIECE_TO_IDX, SQUARE_TO_IDX, IDX_TO_UCI_MOVE
 except:
-    from .mappings import MAX_HALFMOVES, MAX_FULLMOVES, EMPTY_SQ_IDX, PIECE_TO_IDX, SQUARE_TO_IDX, IDX_TO_UCI_MOVE
 # --- Tokenizer --- #
 class FENTokenizer(nn.Module):

 from huggingface_hub import PyTorchModelHubMixin
 try:
+    from mapping import MAX_HALFMOVES, MAX_FULLMOVES, EMPTY_SQ_IDX, PIECE_TO_IDX, SQUARE_TO_IDX, IDX_TO_UCI_MOVE
 except:
+    from .mapping import MAX_HALFMOVES, MAX_FULLMOVES, EMPTY_SQ_IDX, PIECE_TO_IDX, SQUARE_TO_IDX, IDX_TO_UCI_MOVE
 # --- Tokenizer --- #
 class FENTokenizer(nn.Module):