Sushil Singh commited on
Commit ·
2cea1e3
1
Parent(s): 4929634
removed register.py code moved to model
Browse files- register.py +0 -8
register.py
DELETED
|
@@ -1,8 +0,0 @@
|
|
| 1 |
-
from transformers import AutoConfig, AutoModel
|
| 2 |
-
from model import SimpleLinearConfig, SimpleLinearPreTrainedModel
|
| 3 |
-
|
| 4 |
-
# Register our config class with AutoConfig
|
| 5 |
-
AutoConfig.register("simple-linear-model", SimpleLinearConfig)
|
| 6 |
-
|
| 7 |
-
# Register our model class with AutoModel
|
| 8 |
-
AutoModel.register(SimpleLinearConfig, SimpleLinearPreTrainedModel)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|