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
Files changed (2) hide show
  1. config.json +2 -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