Commit
•
6e0776a
1
Parent(s):
0b35b1b
Update configuration_fake_custom.py
Browse files
configuration_fake_custom.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
from transformers import
|
2 |
|
3 |
|
4 |
-
class FakeCustomConfig(
|
5 |
-
model_type = "fakecustom"
|
|
|
1 |
+
from transformers import PretrainedConfig
|
2 |
|
3 |
|
4 |
+
class FakeCustomConfig(PretrainedConfig):
|
5 |
+
model_type = "fakecustom"
|