update

model_index.json

@@ -1,7 +1,16 @@
 {
-  "_class_name": "KandinskyV3Pipeline",
-  "_diffusers_version": "0.17.0.dev0",
-  "text_encoder":[
+  "_class_name": "Kandinsky3Pipeline",
+  "_diffusers_version": "0.24.0.dev0",
+  "_name_or_path": "/home/patrick/kandinsky-3",
+  "movq": [
+    "diffusers",
+    "VQModel"
+  ],
+  "scheduler": [
+    "diffusers",
+    "DDPMScheduler"
+  ],
+  "text_encoder": [
     "transformers",
     "T5EncoderModel"
   ],
@@ -11,14 +20,6 @@
   ],
   "unet": [
     "diffusers",
-    "UNetKandi3"
-  ],
-  "scheduler": [
-    "diffusers",
-    "DDPMScheduler"
-  ],
-   "movq": [
-    "diffusers",
-    "VQModel"
+    "Kandinsky3UNet"
   ]
-}
+}

movq/config.json

@@ -1,6 +1,7 @@
 {
   "_class_name": "VQModel",
-  "_diffusers_version": "0.17.0.dev0",
+  "_diffusers_version": "0.24.0.dev0",
+  "_name_or_path": "/home/patrick/kandinsky-3/movq",
   "act_fn": "silu",
   "block_out_channels": [
     256,
@@ -30,4 +31,4 @@
     "UpDecoderBlock2D"
   ],
   "vq_embed_dim": 4
-}
+}

movq/{diffusion_pytorch_model.bin → diffusion_pytorch_model.fp16.safetensors}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:53fef29c42ce1e01c25c492b44a08485f95998e7f64397ae09e057c9b747b745
-size 1082846259
+oid sha256:8a4705625c160c49025e1f9b98776aaf5287240ca0d5ed39a1a235598fc39244
+size 541388942

scheduler/scheduler_config.json

@@ -1,13 +1,19 @@
 {
   "_class_name": "DDPMScheduler",
-  "_diffusers_version": "0.17.0.dev0",
-  "num_train_timesteps": 1000,
-  "beta_schedule":  "squaredcos_cap_v2",
+  "_diffusers_version": "0.24.0.dev0",
+  "beta_end": 0.012,
+  "beta_schedule": "squaredcos_cap_v2",
   "beta_start": 0.00085,
-  "beta_end":0.012,
-  "clip_sample" : true,
-  "set_alpha_to_one" : false, 
-  "steps_offset" : 1,
-  "prediction_type" : "epsilon",
-  "thresholding" : false
-}
+  "clip_sample": true,
+  "clip_sample_range": 1.0,
+  "dynamic_thresholding_ratio": 0.995,
+  "num_train_timesteps": 1000,
+  "prediction_type": "epsilon",
+  "sample_max_value": 1.0,
+  "set_alpha_to_one": false,
+  "steps_offset": 1,
+  "thresholding": false,
+  "timestep_spacing": "leading",
+  "trained_betas": null,
+  "variance_type": "fixed_small"
+}

text_encoder/config.json

@@ -1,8 +1,9 @@
 {
-  "_name_or_path": "/home/younes_huggingface_co/code/ul2/flan-ul2",
+  "_name_or_path": "/home/patrick/kandinsky-3/text_encoder",
   "architectures": [
-    "T5ForConditionalGeneration"
+    "T5EncoderModel"
   ],
+  "classifier_dropout": 0.0,
   "d_ff": 16384,
   "d_kv": 256,
   "d_model": 4096,
@@ -24,8 +25,8 @@
   "pad_token_id": 0,
   "relative_attention_max_distance": 128,
   "relative_attention_num_buckets": 32,
-  "torch_dtype": "bfloat16",
-  "transformers_version": "4.25.0.dev0",
+  "torch_dtype": "float16",
+  "transformers_version": "4.36.0.dev0",
   "use_cache": true,
   "vocab_size": 32128
 }

text_encoder/{pytorch_model.bin → model.fp16-00001-of-00005.safetensors}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b804e13439715df6fc1f7b5bd602df7aeef06bd3c2880862597e83f38a8d7c0c
-size 34887326513
+oid sha256:2745f08c74c13536c67282afc1e5ba80daf67e5ded7925845f3cb5e1983a2826
+size 4994491352

unet/diffusion_pytorch_model.bin → text_encoder/model.fp16-00002-of-00005.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a1986f3486c37bc0e9e91ab823ba21de270fb8bb41344d7c5485bc27f9945826
-size 12221796223
+oid sha256:a1c184e0f3a2e8c80360379ada2d9fce40bee23647a63e3baefa00d837084408
+size 4932632832

text_encoder/model.fp16-00003-of-00005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a6e4c548e5c2e6aa080a5eefad31e973dbba55b6918f95fb77af97fce341c94
+size 4831960880

text_encoder/model.fp16-00004-of-00005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e7652beb11f33c5549d391c2affacd6642d9ca21bc0d4319cba60fe1b357ff0
+size 4999741472

text_encoder/model.fp16-00005-of-00005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f41b9aea886f3d583c026011cb7251ff21f615774ad604264aaaa01605e02326
+size 1979772184

text_encoder/model.safetensors.index.fp16.json

@@ -0,0 +1,298 @@
+{
+  "metadata": {
+    "total_size": 21738562560
+  },
+  "weight_map": {
+    "encoder.block.0.layer.0.SelfAttention.k.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.o.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.q.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.relative_attention_bias.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.v.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.0.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wo.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.0.layer.1.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.k.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.o.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.q.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.v.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.0.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wo.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.1.layer.1.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.k.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.o.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.q.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.v.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.10.layer.0.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wo.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.10.layer.1.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.k.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.o.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.q.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.v.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.0.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wo.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.11.layer.1.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.k.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.o.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.q.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.v.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.0.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wo.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.12.layer.1.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.k.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.o.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.q.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.v.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.0.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wo.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.13.layer.1.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.k.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.o.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.q.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.v.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.14.layer.0.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wo.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.14.layer.1.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.k.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.o.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.q.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.v.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.0.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wo.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.15.layer.1.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.k.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.o.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.q.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.v.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.0.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wo.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.16.layer.1.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.k.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.o.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.q.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.v.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.0.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wo.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.17.layer.1.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.k.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.o.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.q.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.v.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.0.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wo.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.18.layer.1.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.k.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.o.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.q.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.v.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.0.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wo.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.19.layer.1.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.k.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.o.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.q.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.v.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.2.layer.0.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wo.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.2.layer.1.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.k.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.o.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.q.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.v.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.20.layer.0.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wo.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.20.layer.1.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.k.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.o.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.q.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.v.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.21.layer.0.layer_norm.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00003-of-00005.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wo.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.21.layer.1.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.k.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.o.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.q.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.v.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.0.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wo.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.22.layer.1.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.k.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.o.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.q.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.v.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.0.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wo.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.23.layer.1.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.0.SelfAttention.k.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.0.SelfAttention.o.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.0.SelfAttention.q.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.0.SelfAttention.v.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.0.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.1.DenseReluDense.wo.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.24.layer.1.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.0.SelfAttention.k.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.0.SelfAttention.o.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.0.SelfAttention.q.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.0.SelfAttention.v.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.0.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.1.DenseReluDense.wo.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.25.layer.1.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.0.SelfAttention.k.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.0.SelfAttention.o.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.0.SelfAttention.q.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.0.SelfAttention.v.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.0.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.1.DenseReluDense.wo.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.26.layer.1.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.0.SelfAttention.k.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.0.SelfAttention.o.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.0.SelfAttention.q.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.0.SelfAttention.v.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.0.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.1.DenseReluDense.wo.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.27.layer.1.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.0.SelfAttention.k.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.0.SelfAttention.o.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.0.SelfAttention.q.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.0.SelfAttention.v.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.0.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.1.DenseReluDense.wo.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.28.layer.1.layer_norm.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.29.layer.0.SelfAttention.k.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.29.layer.0.SelfAttention.o.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.29.layer.0.SelfAttention.q.weight": "model.fp16-00004-of-00005.safetensors",
+    "encoder.block.29.layer.0.SelfAttention.v.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.29.layer.0.layer_norm.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.29.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.29.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.29.layer.1.DenseReluDense.wo.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.29.layer.1.layer_norm.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.k.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.o.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.q.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.v.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.3.layer.0.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wo.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.3.layer.1.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.30.layer.0.SelfAttention.k.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.30.layer.0.SelfAttention.o.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.30.layer.0.SelfAttention.q.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.30.layer.0.SelfAttention.v.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.30.layer.0.layer_norm.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.30.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.30.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.30.layer.1.DenseReluDense.wo.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.30.layer.1.layer_norm.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.0.SelfAttention.k.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.0.SelfAttention.o.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.0.SelfAttention.q.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.0.SelfAttention.v.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.0.layer_norm.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.1.DenseReluDense.wo.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.31.layer.1.layer_norm.weight": "model.fp16-00005-of-00005.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.k.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.o.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.q.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.v.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.4.layer.0.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wo.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.4.layer.1.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.k.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.o.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.q.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.v.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.0.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wo.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.5.layer.1.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.k.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.o.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.q.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.v.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.0.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wo.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.6.layer.1.layer_norm.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.k.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.o.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.q.weight": "model.fp16-00001-of-00005.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.v.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.7.layer.0.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wo.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.7.layer.1.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.k.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.o.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.q.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.v.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.0.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wo.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.8.layer.1.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.k.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.o.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.q.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.v.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.0.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wi_0.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wi_1.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wo.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.block.9.layer.1.layer_norm.weight": "model.fp16-00002-of-00005.safetensors",
+    "encoder.final_layer_norm.weight": "model.fp16-00005-of-00005.safetensors",
+    "shared.weight": "model.fp16-00001-of-00005.safetensors"
+  }
+}

tokenizer/added_tokens.json

@@ -1,4 +1,104 @@
 {
+  "<extra_id_0>": 32099,
+  "<extra_id_10>": 32089,
+  "<extra_id_11>": 32088,
+  "<extra_id_12>": 32087,
+  "<extra_id_13>": 32086,
+  "<extra_id_14>": 32085,
+  "<extra_id_15>": 32084,
+  "<extra_id_16>": 32083,
+  "<extra_id_17>": 32082,
+  "<extra_id_18>": 32081,
+  "<extra_id_19>": 32080,
+  "<extra_id_1>": 32098,
+  "<extra_id_20>": 32079,
+  "<extra_id_21>": 32078,
+  "<extra_id_22>": 32077,
+  "<extra_id_23>": 32076,
+  "<extra_id_24>": 32075,
+  "<extra_id_25>": 32074,
+  "<extra_id_26>": 32073,
+  "<extra_id_27>": 32072,
+  "<extra_id_28>": 32071,
+  "<extra_id_29>": 32070,
+  "<extra_id_2>": 32097,
+  "<extra_id_30>": 32069,
+  "<extra_id_31>": 32068,
+  "<extra_id_32>": 32067,
+  "<extra_id_33>": 32066,
+  "<extra_id_34>": 32065,
+  "<extra_id_35>": 32064,
+  "<extra_id_36>": 32063,
+  "<extra_id_37>": 32062,
+  "<extra_id_38>": 32061,
+  "<extra_id_39>": 32060,
+  "<extra_id_3>": 32096,
+  "<extra_id_40>": 32059,
+  "<extra_id_41>": 32058,
+  "<extra_id_42>": 32057,
+  "<extra_id_43>": 32056,
+  "<extra_id_44>": 32055,
+  "<extra_id_45>": 32054,
+  "<extra_id_46>": 32053,
+  "<extra_id_47>": 32052,
+  "<extra_id_48>": 32051,
+  "<extra_id_49>": 32050,
+  "<extra_id_4>": 32095,
+  "<extra_id_50>": 32049,
+  "<extra_id_51>": 32048,
+  "<extra_id_52>": 32047,
+  "<extra_id_53>": 32046,
+  "<extra_id_54>": 32045,
+  "<extra_id_55>": 32044,
+  "<extra_id_56>": 32043,
+  "<extra_id_57>": 32042,
+  "<extra_id_58>": 32041,
+  "<extra_id_59>": 32040,
+  "<extra_id_5>": 32094,
+  "<extra_id_60>": 32039,
+  "<extra_id_61>": 32038,
+  "<extra_id_62>": 32037,
+  "<extra_id_63>": 32036,
+  "<extra_id_64>": 32035,
+  "<extra_id_65>": 32034,
+  "<extra_id_66>": 32033,
+  "<extra_id_67>": 32032,
+  "<extra_id_68>": 32031,
+  "<extra_id_69>": 32030,
+  "<extra_id_6>": 32093,
+  "<extra_id_70>": 32029,
+  "<extra_id_71>": 32028,
+  "<extra_id_72>": 32027,
+  "<extra_id_73>": 32026,
+  "<extra_id_74>": 32025,
+  "<extra_id_75>": 32024,
+  "<extra_id_76>": 32023,
+  "<extra_id_77>": 32022,
+  "<extra_id_78>": 32021,
+  "<extra_id_79>": 32020,
+  "<extra_id_7>": 32092,
+  "<extra_id_80>": 32019,
+  "<extra_id_81>": 32018,
+  "<extra_id_82>": 32017,
+  "<extra_id_83>": 32016,
+  "<extra_id_84>": 32015,
+  "<extra_id_85>": 32014,
+  "<extra_id_86>": 32013,
+  "<extra_id_87>": 32012,
+  "<extra_id_88>": 32011,
+  "<extra_id_89>": 32010,
+  "<extra_id_8>": 32091,
+  "<extra_id_90>": 32009,
+  "<extra_id_91>": 32008,
+  "<extra_id_92>": 32007,
+  "<extra_id_93>": 32006,
+  "<extra_id_94>": 32005,
+  "<extra_id_95>": 32004,
+  "<extra_id_96>": 32003,
+  "<extra_id_97>": 32002,
+  "<extra_id_98>": 32001,
+  "<extra_id_99>": 32000,
+  "<extra_id_9>": 32090,
   "[new_id_0]": 32100,
   "[new_id_10]": 32110,
   "[new_id_11]": 32111,

tokenizer/special_tokens_map.json

@@ -101,7 +101,25 @@
     "<extra_id_98>",
     "<extra_id_99>"
   ],
-  "eos_token": "</s>",
-  "pad_token": "<pad>",
-  "unk_token": "<unk>"
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
 }

tokenizer/tokenizer_config.json

@@ -1,4 +1,1054 @@
 {
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32000": {
+      "content": "<extra_id_99>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32001": {
+      "content": "<extra_id_98>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32002": {
+      "content": "<extra_id_97>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32003": {
+      "content": "<extra_id_96>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32004": {
+      "content": "<extra_id_95>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32005": {
+      "content": "<extra_id_94>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32006": {
+      "content": "<extra_id_93>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32007": {
+      "content": "<extra_id_92>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32008": {
+      "content": "<extra_id_91>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32009": {
+      "content": "<extra_id_90>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32010": {
+      "content": "<extra_id_89>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32011": {
+      "content": "<extra_id_88>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32012": {
+      "content": "<extra_id_87>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32013": {
+      "content": "<extra_id_86>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32014": {
+      "content": "<extra_id_85>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32015": {
+      "content": "<extra_id_84>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32016": {
+      "content": "<extra_id_83>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32017": {
+      "content": "<extra_id_82>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32018": {
+      "content": "<extra_id_81>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32019": {
+      "content": "<extra_id_80>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32020": {
+      "content": "<extra_id_79>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32021": {
+      "content": "<extra_id_78>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32022": {
+      "content": "<extra_id_77>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32023": {
+      "content": "<extra_id_76>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32024": {
+      "content": "<extra_id_75>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32025": {
+      "content": "<extra_id_74>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32026": {
+      "content": "<extra_id_73>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32027": {
+      "content": "<extra_id_72>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32028": {
+      "content": "<extra_id_71>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32029": {
+      "content": "<extra_id_70>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32030": {
+      "content": "<extra_id_69>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32031": {
+      "content": "<extra_id_68>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32032": {
+      "content": "<extra_id_67>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32033": {
+      "content": "<extra_id_66>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32034": {
+      "content": "<extra_id_65>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32035": {
+      "content": "<extra_id_64>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32036": {
+      "content": "<extra_id_63>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32037": {
+      "content": "<extra_id_62>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32038": {
+      "content": "<extra_id_61>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32039": {
+      "content": "<extra_id_60>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32040": {
+      "content": "<extra_id_59>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32041": {
+      "content": "<extra_id_58>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32042": {
+      "content": "<extra_id_57>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32043": {
+      "content": "<extra_id_56>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32044": {
+      "content": "<extra_id_55>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32045": {
+      "content": "<extra_id_54>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32046": {
+      "content": "<extra_id_53>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32047": {
+      "content": "<extra_id_52>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32048": {
+      "content": "<extra_id_51>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32049": {
+      "content": "<extra_id_50>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32050": {
+      "content": "<extra_id_49>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32051": {
+      "content": "<extra_id_48>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32052": {
+      "content": "<extra_id_47>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32053": {
+      "content": "<extra_id_46>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32054": {
+      "content": "<extra_id_45>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32055": {
+      "content": "<extra_id_44>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32056": {
+      "content": "<extra_id_43>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32057": {
+      "content": "<extra_id_42>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32058": {
+      "content": "<extra_id_41>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32059": {
+      "content": "<extra_id_40>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32060": {
+      "content": "<extra_id_39>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32061": {
+      "content": "<extra_id_38>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32062": {
+      "content": "<extra_id_37>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32063": {
+      "content": "<extra_id_36>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32064": {
+      "content": "<extra_id_35>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32065": {
+      "content": "<extra_id_34>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32066": {
+      "content": "<extra_id_33>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32067": {
+      "content": "<extra_id_32>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32068": {
+      "content": "<extra_id_31>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32069": {
+      "content": "<extra_id_30>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32070": {
+      "content": "<extra_id_29>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32071": {
+      "content": "<extra_id_28>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32072": {
+      "content": "<extra_id_27>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32073": {
+      "content": "<extra_id_26>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32074": {
+      "content": "<extra_id_25>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32075": {
+      "content": "<extra_id_24>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32076": {
+      "content": "<extra_id_23>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32077": {
+      "content": "<extra_id_22>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32078": {
+      "content": "<extra_id_21>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32079": {
+      "content": "<extra_id_20>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32080": {
+      "content": "<extra_id_19>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32081": {
+      "content": "<extra_id_18>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32082": {
+      "content": "<extra_id_17>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32083": {
+      "content": "<extra_id_16>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32084": {
+      "content": "<extra_id_15>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32085": {
+      "content": "<extra_id_14>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32086": {
+      "content": "<extra_id_13>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32087": {
+      "content": "<extra_id_12>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32088": {
+      "content": "<extra_id_11>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32089": {
+      "content": "<extra_id_10>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32090": {
+      "content": "<extra_id_9>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32091": {
+      "content": "<extra_id_8>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32092": {
+      "content": "<extra_id_7>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32093": {
+      "content": "<extra_id_6>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32094": {
+      "content": "<extra_id_5>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32095": {
+      "content": "<extra_id_4>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32096": {
+      "content": "<extra_id_3>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32097": {
+      "content": "<extra_id_2>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32098": {
+      "content": "<extra_id_1>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32099": {
+      "content": "<extra_id_0>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": true,
+      "special": true
+    },
+    "32100": {
+      "content": "[new_id_0]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32101": {
+      "content": "[new_id_1]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32102": {
+      "content": "[new_id_2]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32103": {
+      "content": "[new_id_3]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32104": {
+      "content": "[new_id_4]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32105": {
+      "content": "[new_id_5]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32106": {
+      "content": "[new_id_6]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32107": {
+      "content": "[new_id_7]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32108": {
+      "content": "[new_id_8]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32109": {
+      "content": "[new_id_9]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32110": {
+      "content": "[new_id_10]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32111": {
+      "content": "[new_id_11]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32112": {
+      "content": "[new_id_12]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32113": {
+      "content": "[new_id_13]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32114": {
+      "content": "[new_id_14]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32115": {
+      "content": "[new_id_15]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32116": {
+      "content": "[new_id_16]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32117": {
+      "content": "[new_id_17]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32118": {
+      "content": "[new_id_18]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32119": {
+      "content": "[new_id_19]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32120": {
+      "content": "[new_id_20]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32121": {
+      "content": "[new_id_21]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32122": {
+      "content": "[new_id_22]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32123": {
+      "content": "[new_id_23]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32124": {
+      "content": "[new_id_24]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32125": {
+      "content": "[new_id_25]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32126": {
+      "content": "[new_id_26]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "32127": {
+      "content": "[new_id_27]",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
   "additional_special_tokens": [
     "<extra_id_0>",
     "<extra_id_1>",
@@ -104,6 +1154,7 @@
   "clean_up_tokenization_spaces": true,
   "eos_token": "</s>",
   "extra_ids": 100,
+  "legacy": true,
   "model_max_length": 2048,
   "pad_token": "<pad>",
   "sp_model_kwargs": {},

unet/config.json

@@ -1,18 +1,17 @@
- {
-   "_class_name": "UNet2DConditionModel",
-   "_diffusers_version": "0.17.0.dev0",
-   "model_channels": 384,
-   "num_channels": 4,
-   "init_channels": 192,
-   "time_embed_dim": 1536,
-   "context_dim": 4096,
-   "model_dim": 4096,
-   "groups": 32,
-   "head_dim": 64,
-   "expansion_ratio": 4,
-   "compression_ratio": 2,
-   "dim_mult": [1, 2, 4, 8],
-   "num_blocks": [3, 3, 3, 3],
-   "add_cross_attention": [false, true, true, true],
-   "add_self_attention": [false, true, true, true]
-}
+{
+  "_class_name": "Kandinsky3UNet",
+  "_diffusers_version": "0.24.0.dev0",
+  "attention_head_dim": 64,
+  "block_out_channels": [
+    384,
+    768,
+    1536,
+    3072
+  ],
+  "cross_attention_dim": 4096,
+  "encoder_hid_dim": 4096,
+  "groups": 32,
+  "in_channels": 4,
+  "layers_per_block": 3,
+  "time_embedding_dim": 1536
+}

unet/diffusion_pytorch_model.fp16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:805179fe9c10cd1e01c59057f0c2bb07ce57834ad343f721e50ca16bef6952f6
+size 6110842296