matheoqtb commited on
Commit
b108f23
·
verified ·
1 Parent(s): 453a645

Upload __init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. __init__.py +22 -0
__init__.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Chess Challenge source module."""
2
+
3
+ from .model import ChessConfig, ChessForCausalLM
4
+ from .tokenizer import ChessTokenizer
5
+
6
+ # Lazy import for evaluate to avoid RuntimeWarning when running as module
7
+ def __getattr__(name):
8
+ if name == "ChessEvaluator":
9
+ from .evaluate import ChessEvaluator
10
+ return ChessEvaluator
11
+ if name == "load_model_from_hub":
12
+ from .evaluate import load_model_from_hub
13
+ return load_model_from_hub
14
+ raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
15
+
16
+ __all__ = [
17
+ "ChessConfig",
18
+ "ChessForCausalLM",
19
+ "ChessTokenizer",
20
+ "ChessEvaluator",
21
+ "load_model_from_hub",
22
+ ]