add model

Files changed (6) hide show

config.json CHANGED Viewed

@@ -10,6 +10,9 @@
     null,
     null
   ],
   "attn_dropout": 0.0,
   "attn_init_scale": 1.0,
   "attn_order": [
@@ -197,6 +200,7 @@
   "summary_type": "cls_index",
   "summary_use_proj": true,
   "t_bins": 128,
   "transformers_version": "4.21.0.dev0",
   "use_cache": true,
   "use_nonrelative_specloss": true,
@@ -236,13 +240,13 @@
   "vq_vae_width": 64,
   "vqvae_z_shapes": [
     [
-      125
     ],
     [
-      500
     ],
     [
-      2000
     ]
   ],
   "width": [

     null,
     null
   ],
+  "architectures": [
+    "JukeboxModel"
+  ],
   "attn_dropout": 0.0,
   "attn_init_scale": 1.0,
   "attn_order": [
   "summary_type": "cls_index",
   "summary_use_proj": true,
   "t_bins": 128,
+  "torch_dtype": "float32",
   "transformers_version": "4.21.0.dev0",
   "use_cache": true,
   "use_nonrelative_specloss": true,
   "vq_vae_width": 64,
   "vqvae_z_shapes": [
     [
+      344
     ],
     [
+      1378
     ],
     [
+      5512
     ]
   ],
   "width": [

pytorch_model-00001-of-00004.bin ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:1888fd7bfbde3e167e7e2928d86298aa21232c2a7c9a60dec72c1c50e7530cd3
+size 9911092875

pytorch_model-00002-of-00004.bin ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:7647b989cd65f445c6bc2e03a145ae14cfce5972f6dba2d254295af219d68b4e
+size 9958784493

pytorch_model-00003-of-00004.bin ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:be0c105293bcb6d1afc43aa525523273a5e1209e899df3f47a2866275a8a8275
+size 9958784533

pytorch_model-00004-of-00004.bin ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:c9f87a98f1ef746626333007f2dd31ce1d8624118d0ebc2bf1cd316f847d4293
+size 592587348

pytorch_model.bin.index.json ADDED Viewed

The diff for this file is too large to render. See raw diff