skatzR commited on
Commit
825c9d4
·
verified ·
1 Parent(s): c3a2b52

Create __init__.py

Browse files
Files changed (1) hide show
  1. __init__.py +8 -0
__init__.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from .modeling_rqa import RQAModelConfig, RQAModelHF
2
+
3
+ __all__ = ["RQAModelConfig", "RQAModelHF"]
4
+
5
+ from transformers import AutoConfig, AutoModel
6
+
7
+ AutoConfig.register("rqa", RQAModelConfig)
8
+ AutoModel.register(RQAModelConfig, RQAModelHF)