Upload model

config.json

@@ -1,6 +1,6 @@
 {
   "_commit_hash": null,
-  "_name_or_path": "/home/yoach/bark_weigths/model/",
+  "_name_or_path": "/home/yoach/bark_weigths/model_small/",
   "architectures": [
     "BarkModel"
   ],
@@ -28,7 +28,7 @@
     "finetuning_task": null,
     "forced_bos_token_id": null,
     "forced_eos_token_id": null,
-    "hidden_size": 1024,
+    "hidden_size": 768,
     "id2label": {
       "0": "LABEL_0",
       "1": "LABEL_1"
@@ -50,8 +50,8 @@
     "no_repeat_ngram_size": 0,
     "num_beam_groups": 1,
     "num_beams": 1,
-    "num_heads": 16,
-    "num_layers": 24,
+    "num_heads": 12,
+    "num_layers": 12,
     "num_return_sequences": 1,
     "output_attentions": false,
     "output_hidden_states": false,
@@ -205,7 +205,7 @@
     "finetuning_task": null,
     "forced_bos_token_id": null,
     "forced_eos_token_id": null,
-    "hidden_size": 1024,
+    "hidden_size": 768,
     "id2label": {
       "0": "LABEL_0",
       "1": "LABEL_1"
@@ -227,8 +227,8 @@
     "no_repeat_ngram_size": 0,
     "num_beam_groups": 1,
     "num_beams": 1,
-    "num_heads": 16,
-    "num_layers": 24,
+    "num_heads": 12,
+    "num_layers": 12,
     "num_return_sequences": 1,
     "output_attentions": false,
     "output_hidden_states": false,
@@ -285,7 +285,7 @@
     "finetuning_task": null,
     "forced_bos_token_id": null,
     "forced_eos_token_id": null,
-    "hidden_size": 1024,
+    "hidden_size": 768,
     "id2label": {
       "0": "LABEL_0",
       "1": "LABEL_1"
@@ -307,8 +307,8 @@
     "no_repeat_ngram_size": 0,
     "num_beam_groups": 1,
     "num_beams": 1,
-    "num_heads": 16,
-    "num_layers": 24,
+    "num_heads": 12,
+    "num_layers": 12,
     "num_return_sequences": 1,
     "output_attentions": false,
     "output_hidden_states": false,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d9ac007f13624882c82d561378eb356321594eabf320fdbbe32c25b098dad4df
-size 4516921809
+oid sha256:dc221dd06f22eb8c6babc17f3d2aaae90dbe4223c7dda46b0a62e2c820bbdabd
+size 1699372613