add config

config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "ArthurZ/jukebox-5b-lyrics",
   "activation_function": "quick_gelu",
   "alignment_head": [
     2,
@@ -65,6 +66,11 @@
   "emb_dropout": 0.0,
   "eos_token_id": 50256,
   "fp16_params": true,
+  "hop_fraction": [
+    0.125,
+    0.5,
+    0.5
+  ],
   "hop_length": 256,
   "init_scale": [
     0.1,
@@ -117,6 +123,7 @@
   "n_inner": null,
   "n_layer": 12,
   "n_positions": 1024,
+  "n_samples": 1,
   "n_tokens": [
     512,
     0,
@@ -177,8 +184,8 @@
   "res_scale": false,
   "resid_dropout": 0.0,
   "sample_hop_length": 30000,
-  "sample_length": 44100,
-  "sample_length_in_seconds": 1,
+  "sample_length": 1058304,
+  "sample_length_in_seconds": 24,
   "scale_attn_by_inverse_layer_idx": false,
   "scale_attn_weights": true,
   "single_enc_dec": [
@@ -201,6 +208,7 @@
   "summary_use_proj": true,
   "t_bins": 128,
   "torch_dtype": "float32",
+  "total_sample_length_in_seconds": 180,
   "transformers_version": "4.21.0.dev0",
   "use_cache": true,
   "use_nonrelative_specloss": true,
@@ -240,13 +248,13 @@
   "vq_vae_width": 64,
   "vqvae_z_shapes": [
     [
-      344
+      2067
     ],
     [
-      1378
+      8268
     ],
     [
-      5512
+      33072
     ]
   ],
   "width": [