Vittorio Pippi
commited on
Commit
·
2f623a1
1
Parent(s):
d577eed
Update paths in config.json for AutoConfig and AutoModel; add type hint for config parameter in Emuru model initializer
Browse files- config.json +2 -2
- modeling_emuru.py +1 -1
config.json
CHANGED
@@ -4,8 +4,8 @@
|
|
4 |
"Emuru"
|
5 |
],
|
6 |
"auto_map": {
|
7 |
-
"AutoConfig": "configuration_emuru.EmuruConfig",
|
8 |
-
"AutoModel": "modeling_emuru.Emuru"
|
9 |
},
|
10 |
"model_type": "emuru",
|
11 |
"slices_per_query": 1,
|
|
|
4 |
"Emuru"
|
5 |
],
|
6 |
"auto_map": {
|
7 |
+
"AutoConfig": "blowing-up-groundhogs/emuru--configuration_emuru.EmuruConfig",
|
8 |
+
"AutoModel": "blowing-up-groundhogs/emuru--modeling_emuru.Emuru"
|
9 |
},
|
10 |
"model_type": "emuru",
|
11 |
"slices_per_query": 1,
|
modeling_emuru.py
CHANGED
@@ -30,7 +30,7 @@ class Emuru(PreTrainedModel):
|
|
30 |
"""
|
31 |
config_class = EmuruConfig
|
32 |
|
33 |
-
def __init__(self, config) -> None:
|
34 |
"""
|
35 |
Initialize the Emuru model.
|
36 |
|
|
|
30 |
"""
|
31 |
config_class = EmuruConfig
|
32 |
|
33 |
+
def __init__(self, config: EmuruConfig) -> None:
|
34 |
"""
|
35 |
Initialize the Emuru model.
|
36 |
|