lklimkiewicz commited on
Commit
18b1a72
1 Parent(s): de022db

Upload OPTForMusicGeneration

Browse files
Files changed (3) hide show
  1. config.json +2 -2
  2. model.py +0 -1
  3. model.safetensors +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/home/lklimkiewicz/priv/midi/src/new_model",
3
  "_remove_final_layer_norm": false,
4
  "activation_dropout": 0.0,
5
  "activation_function": "relu",
@@ -8,7 +8,7 @@
8
  ],
9
  "attention_dropout": 0.0,
10
  "auto_map": {
11
- "AutoModelForCausalLM": "model.OPTForMusicGeneration"
12
  },
13
  "bos_token_id": 1,
14
  "do_layer_norm_before": true,
 
1
  {
2
+ "_name_or_path": "/home/lklimkiewicz/priv/midi/src/test_trainer_1/checkpoint-117000",
3
  "_remove_final_layer_norm": false,
4
  "activation_dropout": 0.0,
5
  "activation_function": "relu",
 
8
  ],
9
  "attention_dropout": 0.0,
10
  "auto_map": {
11
+ "AutoModel": "model.OPTForMusicGeneration"
12
  },
13
  "bos_token_id": 1,
14
  "do_layer_norm_before": true,
model.py CHANGED
@@ -17,4 +17,3 @@ class OPTForMusicGeneration(OPTForCausalLM):
17
 
18
 
19
  OPTForMusicGeneration.register_for_auto_class("AutoModel")
20
- OPTForMusicGeneration.register_for_auto_class("AutoModelForCausalLM")
 
17
 
18
 
19
  OPTForMusicGeneration.register_for_auto_class("AutoModel")
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:248379eb7a6307c3f597280dcfbb920f44f763acad5904b4b16f1ad89bc0badb
3
  size 438704008
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b42210c965e2c81fc44e3925a2be0ef10893b93b33c5247f86479c282e52a8cf
3
  size 438704008