flavioschneider
commited on
Commit
•
3548637
1
Parent(s):
a1ace74
Upload config
Browse files- config.json +7 -0
- dmae_config.py +5 -0
config.json
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"auto_map": {
|
3 |
+
"AutoConfig": "dmae_config.DMAE1dConfig"
|
4 |
+
},
|
5 |
+
"model_type": "archinetai/dmae1d-ATC64-v2",
|
6 |
+
"transformers_version": "4.24.0"
|
7 |
+
}
|
dmae_config.py
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import PretrainedConfig
|
2 |
+
from typing import Sequence
|
3 |
+
|
4 |
+
class DMAE1dConfig(PretrainedConfig):
|
5 |
+
model_type = "archinetai/dmae1d-ATC64-v2"
|