add model

config.json

@@ -11,6 +11,9 @@
     null,
     null
   ],
+  "architectures": [
+    "JukeboxModel"
+  ],
   "attn_dropout": 0.0,
   "attn_init_scale": 1.0,
   "attn_order": [
@@ -42,7 +45,7 @@
     3
   ],
   "cond_m_conv": 1,
-  "cond_res_scale": false,
+  "cond_res_scale": true,
   "cond_width": [
     128,
     1024,
@@ -204,6 +207,7 @@
   "summary_type": "cls_index",
   "summary_use_proj": true,
   "t_bins": 64,
+  "torch_dtype": "float32",
   "total_sample_length_in_seconds": 180,
   "transformers_version": "4.21.0.dev0",
   "use_cache": true,
@@ -244,13 +248,13 @@
   "vq_vae_width": 32,
   "vqvae_z_shapes": [
     [
-      8192
+      8268
     ],
     [
-      32768
+      33072
     ],
     [
-      131072
+      132288
     ]
   ],
   "width": [

pytorch_model-00001-of-00002.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6f0ec227a627bc157254a24f7c29aa5df2db0c85a818c224a1aac8d024d8f5dc
-size 9966514355
+oid sha256:790045f95dc8fe401b8c655de206d9b96957b3427b40313e87094520a489e63f
+size 9966514611

pytorch_model.bin.index.json

@@ -5,136 +5,136 @@
   "weight_map": {
     "priors.0.conditioner_blocks.0.cond.model.0.bias": "pytorch_model-00001-of-00002.bin",
     "priors.0.conditioner_blocks.0.cond.model.0.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.0.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.0.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.0.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.0.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.1.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.1.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.1.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.1.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.10.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.10.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.10.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.10.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.11.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.11.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.11.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.11.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.12.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.12.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.12.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.12.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.13.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.13.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.13.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.13.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.14.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.14.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.14.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.14.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.15.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.15.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.15.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.15.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.2.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.2.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.2.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.2.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.3.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.3.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.3.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.3.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.4.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.4.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.4.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.4.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.5.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.5.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.5.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.5.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.6.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.6.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.6.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.6.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.7.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.7.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.7.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.7.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.8.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.8.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.8.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.8.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.9.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.9.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.9.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.1.0.model.9.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.0.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.0.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.0.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.0.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.1.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.1.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.1.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.1.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.10.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.10.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.10.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.10.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.11.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.11.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.11.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.11.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.12.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.12.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.12.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.12.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.13.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.13.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.13.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.13.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.14.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.14.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.14.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.14.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.15.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.15.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.15.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.15.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.2.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.2.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.2.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.2.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.3.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.3.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.3.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.3.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.4.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.4.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.4.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.4.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.5.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.5.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.5.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.5.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.6.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.6.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.6.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.6.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.7.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.7.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.7.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.7.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.8.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.8.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.8.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.8.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.9.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.9.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.9.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.1.0.blocks.9.model.3.weight": "pytorch_model-00001-of-00002.bin",
     "priors.0.conditioner_blocks.0.cond.model.1.1.bias": "pytorch_model-00001-of-00002.bin",
     "priors.0.conditioner_blocks.0.cond.model.1.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.0.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.0.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.0.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.0.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.1.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.1.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.1.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.1.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.10.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.10.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.10.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.10.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.11.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.11.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.11.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.11.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.12.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.12.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.12.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.12.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.13.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.13.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.13.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.13.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.14.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.14.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.14.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.14.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.15.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.15.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.15.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.15.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.2.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.2.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.2.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.2.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.3.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.3.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.3.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.3.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.4.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.4.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.4.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.4.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.5.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.5.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.5.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.5.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.6.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.6.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.6.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.6.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.7.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.7.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.7.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.7.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.8.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.8.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.8.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.8.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.9.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.9.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.9.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.0.conditioner_blocks.0.cond.model.2.0.model.9.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.0.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.0.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.0.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.0.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.1.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.1.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.1.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.1.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.10.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.10.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.10.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.10.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.11.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.11.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.11.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.11.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.12.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.12.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.12.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.12.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.13.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.13.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.13.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.13.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.14.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.14.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.14.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.14.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.15.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.15.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.15.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.15.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.2.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.2.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.2.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.2.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.3.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.3.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.3.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.3.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.4.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.4.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.4.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.4.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.5.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.5.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.5.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.5.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.6.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.6.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.6.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.6.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.7.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.7.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.7.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.7.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.8.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.8.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.8.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.8.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.9.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.9.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.9.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.0.conditioner_blocks.0.cond.model.2.0.blocks.9.model.3.weight": "pytorch_model-00001-of-00002.bin",
     "priors.0.conditioner_blocks.0.cond.model.2.1.bias": "pytorch_model-00001-of-00002.bin",
     "priors.0.conditioner_blocks.0.cond.model.2.1.weight": "pytorch_model-00001-of-00002.bin",
     "priors.0.conditioner_blocks.0.ln.bias": "pytorch_model-00001-of-00002.bin",
@@ -1011,136 +1011,136 @@
     "priors.0.y_emb.bow_genre_emb.emb.weight": "pytorch_model-00001-of-00002.bin",
     "priors.1.conditioner_blocks.0.cond.model.0.bias": "pytorch_model-00001-of-00002.bin",
     "priors.1.conditioner_blocks.0.cond.model.0.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.0.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.0.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.0.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.0.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.1.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.1.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.1.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.1.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.10.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.10.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.10.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.10.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.11.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.11.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.11.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.11.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.12.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.12.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.12.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.12.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.13.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.13.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.13.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.13.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.14.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.14.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.14.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.14.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.15.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.15.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.15.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.15.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.2.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.2.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.2.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.2.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.3.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.3.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.3.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.3.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.4.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.4.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.4.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.4.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.5.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.5.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.5.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.5.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.6.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.6.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.6.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.6.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.7.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.7.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.7.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.7.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.8.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.8.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.8.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.8.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.9.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.9.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.9.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.1.0.model.9.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.0.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.0.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.0.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.0.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.1.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.1.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.1.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.1.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.10.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.10.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.10.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.10.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.11.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.11.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.11.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.11.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.12.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.12.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.12.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.12.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.13.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.13.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.13.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.13.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.14.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.14.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.14.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.14.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.15.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.15.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.15.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.15.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.2.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.2.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.2.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.2.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.3.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.3.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.3.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.3.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.4.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.4.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.4.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.4.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.5.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.5.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.5.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.5.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.6.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.6.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.6.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.6.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.7.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.7.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.7.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.7.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.8.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.8.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.8.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.8.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.9.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.9.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.9.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.1.0.blocks.9.model.3.weight": "pytorch_model-00001-of-00002.bin",
     "priors.1.conditioner_blocks.0.cond.model.1.1.bias": "pytorch_model-00001-of-00002.bin",
     "priors.1.conditioner_blocks.0.cond.model.1.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.0.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.0.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.0.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.0.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.1.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.1.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.1.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.1.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.10.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.10.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.10.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.10.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.11.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.11.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.11.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.11.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.12.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.12.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.12.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.12.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.13.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.13.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.13.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.13.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.14.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.14.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.14.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.14.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.15.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.15.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.15.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.15.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.2.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.2.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.2.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.2.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.3.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.3.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.3.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.3.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.4.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.4.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.4.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.4.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.5.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.5.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.5.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.5.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.6.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.6.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.6.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.6.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.7.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.7.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.7.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.7.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.8.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.8.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.8.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.8.model.3.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.9.model.1.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.9.model.1.weight": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.9.model.3.bias": "pytorch_model-00001-of-00002.bin",
-    "priors.1.conditioner_blocks.0.cond.model.2.0.model.9.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.0.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.0.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.0.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.0.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.1.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.1.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.1.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.1.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.10.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.10.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.10.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.10.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.11.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.11.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.11.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.11.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.12.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.12.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.12.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.12.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.13.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.13.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.13.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.13.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.14.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.14.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.14.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.14.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.15.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.15.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.15.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.15.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.2.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.2.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.2.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.2.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.3.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.3.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.3.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.3.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.4.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.4.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.4.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.4.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.5.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.5.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.5.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.5.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.6.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.6.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.6.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.6.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.7.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.7.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.7.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.7.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.8.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.8.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.8.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.8.model.3.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.9.model.1.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.9.model.1.weight": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.9.model.3.bias": "pytorch_model-00001-of-00002.bin",
+    "priors.1.conditioner_blocks.0.cond.model.2.0.blocks.9.model.3.weight": "pytorch_model-00001-of-00002.bin",
     "priors.1.conditioner_blocks.0.cond.model.2.1.bias": "pytorch_model-00001-of-00002.bin",
     "priors.1.conditioner_blocks.0.cond.model.2.1.weight": "pytorch_model-00001-of-00002.bin",
     "priors.1.conditioner_blocks.0.ln.bias": "pytorch_model-00001-of-00002.bin",