Rocketknight1 HF Staff commited on
Commit
5ad4ec9
·
verified ·
1 Parent(s): cc46305

Upload 9 files

Browse files
Files changed (3) hide show
  1. config.json +3 -2
  2. model.safetensors +2 -2
  3. modeling_fake_custom.py +4 -1
config.json CHANGED
@@ -7,12 +7,13 @@
7
  "add_adapter": false,
8
  "apply_spec_augment": true,
9
  "architectures": [
10
- "FakeCustomModel"
11
  ],
12
  "attention_dropout": 0.1,
13
  "auto_map": {
14
  "AutoConfig": "configuration_fake_custom.FakeCustomConfig",
15
- "AutoModel": "modeling_fake_custom.FakeCustomModel"
 
16
  },
17
  "bos_token_id": 1,
18
  "classifier_proj_size": 256,
 
7
  "add_adapter": false,
8
  "apply_spec_augment": true,
9
  "architectures": [
10
+ "FakeCustomModelForCTC"
11
  ],
12
  "attention_dropout": 0.1,
13
  "auto_map": {
14
  "AutoConfig": "configuration_fake_custom.FakeCustomConfig",
15
+ "AutoModel": "modeling_fake_custom.FakeCustomModel",
16
+ "AutoModelForCTC": "modeling_fake_custom.FakeCustomModelForCTC"
17
  },
18
  "bos_token_id": 1,
19
  "classifier_proj_size": 256,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:455cbf5791c30de9bfc004bdda285b8d7dd439ffbab041accd66723812192d7e
3
- size 115096
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:091c1f217d8c688bc769d54489ebb358f7630dddf9bc5abf47d0b42a3b3f64ef
3
+ size 118144
modeling_fake_custom.py CHANGED
@@ -1,5 +1,8 @@
1
- from transformers import Wav2Vec2Model
2
  from .configuration_fake_custom import FakeCustomConfig
3
 
4
  class FakeCustomModel(Wav2Vec2Model):
5
  config_class = FakeCustomConfig
 
 
 
 
1
+ from transformers import Wav2Vec2Model, Wav2Vec2ForCTC
2
  from .configuration_fake_custom import FakeCustomConfig
3
 
4
  class FakeCustomModel(Wav2Vec2Model):
5
  config_class = FakeCustomConfig
6
+
7
+ class FakeCustomModelForCTC(Wav2Vec2ForCTC):
8
+ config_class = FakeCustomConfig