Add model checkpoint

config.json

@@ -0,0 +1,30 @@
+{
+  "_name_or_path": "/root/CS492-Team28-Diffusion-Project-3DVolume/result/112611/checkpoint-11250",
+  "architectures": [
+    "UNet3D"
+  ],
+  "attn_resolutions": [
+    4,
+    8,
+    16
+  ],
+  "ch": 64,
+  "ch_mult": [
+    1,
+    2,
+    4,
+    8,
+    8
+  ],
+  "dropout": 0.1,
+  "image_size": 64,
+  "in_channels": 1,
+  "model_type": "unet3d",
+  "num_classes": 3,
+  "num_diffusion_timesteps": 1000,
+  "num_res_blocks": 2,
+  "out_ch": 1,
+  "resamp_with_conv": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.45.1"
+}

model-00001-of-00013.safetensors

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

model-00002-of-00013.safetensors

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

model-00003-of-00013.safetensors

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

model-00004-of-00013.safetensors

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

model-00005-of-00013.safetensors

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

model-00006-of-00013.safetensors

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

model-00007-of-00013.safetensors

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

model-00008-of-00013.safetensors

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

model-00009-of-00013.safetensors

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

model-00010-of-00013.safetensors

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

model-00011-of-00013.safetensors

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

model-00012-of-00013.safetensors

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

model-00013-of-00013.safetensors

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

model.safetensors.index.json

@@ -0,0 +1,500 @@
+{
+  "metadata": {
+    "total_size": 1121082884
+  },
+  "weight_map": {
+    "class_embed.weight": "model-00001-of-00013.safetensors",
+    "conv_in.bias": "model-00001-of-00013.safetensors",
+    "conv_in.weight": "model-00001-of-00013.safetensors",
+    "conv_out.bias": "model-00013-of-00013.safetensors",
+    "conv_out.weight": "model-00013-of-00013.safetensors",
+    "down.0.block.0.conv1.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.0.conv1.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.0.conv2.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.0.conv2.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.0.norm1.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.0.norm1.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.0.norm2.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.0.norm2.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.0.temb_proj.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.0.temb_proj.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.1.conv1.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.1.conv1.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.1.conv2.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.1.conv2.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.1.norm1.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.1.norm1.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.1.norm2.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.1.norm2.weight": "model-00001-of-00013.safetensors",
+    "down.0.block.1.temb_proj.bias": "model-00001-of-00013.safetensors",
+    "down.0.block.1.temb_proj.weight": "model-00001-of-00013.safetensors",
+    "down.0.downsample.down.bias": "model-00001-of-00013.safetensors",
+    "down.0.downsample.down.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.0.conv1.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.0.conv1.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.0.conv2.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.0.conv2.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.0.norm1.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.0.norm1.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.0.norm2.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.0.norm2.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.0.shortcut.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.0.shortcut.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.0.temb_proj.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.0.temb_proj.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.1.conv1.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.1.conv1.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.1.conv2.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.1.conv2.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.1.norm1.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.1.norm1.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.1.norm2.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.1.norm2.weight": "model-00001-of-00013.safetensors",
+    "down.1.block.1.temb_proj.bias": "model-00001-of-00013.safetensors",
+    "down.1.block.1.temb_proj.weight": "model-00001-of-00013.safetensors",
+    "down.1.downsample.down.bias": "model-00001-of-00013.safetensors",
+    "down.1.downsample.down.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.k.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.k.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.norm.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.norm.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.proj_out.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.proj_out.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.q.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.q.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.v.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.0.v.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.k.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.k.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.norm.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.norm.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.proj_out.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.proj_out.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.q.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.q.weight": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.v.bias": "model-00001-of-00013.safetensors",
+    "down.2.attn.1.v.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.0.conv1.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.0.conv1.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.0.conv2.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.0.conv2.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.0.norm1.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.0.norm1.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.0.norm2.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.0.norm2.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.0.shortcut.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.0.shortcut.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.0.temb_proj.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.0.temb_proj.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.1.conv1.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.1.conv1.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.1.conv2.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.1.conv2.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.1.norm1.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.1.norm1.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.1.norm2.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.1.norm2.weight": "model-00001-of-00013.safetensors",
+    "down.2.block.1.temb_proj.bias": "model-00001-of-00013.safetensors",
+    "down.2.block.1.temb_proj.weight": "model-00001-of-00013.safetensors",
+    "down.2.downsample.down.bias": "model-00001-of-00013.safetensors",
+    "down.2.downsample.down.weight": "model-00001-of-00013.safetensors",
+    "down.3.attn.0.k.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.k.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.norm.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.norm.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.proj_out.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.proj_out.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.q.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.q.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.v.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.0.v.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.k.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.k.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.norm.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.norm.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.proj_out.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.proj_out.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.q.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.q.weight": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.v.bias": "model-00002-of-00013.safetensors",
+    "down.3.attn.1.v.weight": "model-00002-of-00013.safetensors",
+    "down.3.block.0.conv1.bias": "model-00001-of-00013.safetensors",
+    "down.3.block.0.conv1.weight": "model-00001-of-00013.safetensors",
+    "down.3.block.0.conv2.bias": "model-00001-of-00013.safetensors",
+    "down.3.block.0.conv2.weight": "model-00001-of-00013.safetensors",
+    "down.3.block.0.norm1.bias": "model-00001-of-00013.safetensors",
+    "down.3.block.0.norm1.weight": "model-00001-of-00013.safetensors",
+    "down.3.block.0.norm2.bias": "model-00001-of-00013.safetensors",
+    "down.3.block.0.norm2.weight": "model-00001-of-00013.safetensors",
+    "down.3.block.0.shortcut.bias": "model-00001-of-00013.safetensors",
+    "down.3.block.0.shortcut.weight": "model-00001-of-00013.safetensors",
+    "down.3.block.0.temb_proj.bias": "model-00001-of-00013.safetensors",
+    "down.3.block.0.temb_proj.weight": "model-00001-of-00013.safetensors",
+    "down.3.block.1.conv1.bias": "model-00002-of-00013.safetensors",
+    "down.3.block.1.conv1.weight": "model-00002-of-00013.safetensors",
+    "down.3.block.1.conv2.bias": "model-00002-of-00013.safetensors",
+    "down.3.block.1.conv2.weight": "model-00002-of-00013.safetensors",
+    "down.3.block.1.norm1.bias": "model-00001-of-00013.safetensors",
+    "down.3.block.1.norm1.weight": "model-00001-of-00013.safetensors",
+    "down.3.block.1.norm2.bias": "model-00002-of-00013.safetensors",
+    "down.3.block.1.norm2.weight": "model-00002-of-00013.safetensors",
+    "down.3.block.1.temb_proj.bias": "model-00002-of-00013.safetensors",
+    "down.3.block.1.temb_proj.weight": "model-00002-of-00013.safetensors",
+    "down.3.downsample.down.bias": "model-00002-of-00013.safetensors",
+    "down.3.downsample.down.weight": "model-00002-of-00013.safetensors",
+    "down.4.attn.0.k.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.k.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.norm.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.norm.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.proj_out.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.proj_out.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.q.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.q.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.v.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.0.v.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.k.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.k.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.norm.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.norm.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.proj_out.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.proj_out.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.q.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.q.weight": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.v.bias": "model-00004-of-00013.safetensors",
+    "down.4.attn.1.v.weight": "model-00004-of-00013.safetensors",
+    "down.4.block.0.conv1.bias": "model-00003-of-00013.safetensors",
+    "down.4.block.0.conv1.weight": "model-00003-of-00013.safetensors",
+    "down.4.block.0.conv2.bias": "model-00003-of-00013.safetensors",
+    "down.4.block.0.conv2.weight": "model-00003-of-00013.safetensors",
+    "down.4.block.0.norm1.bias": "model-00002-of-00013.safetensors",
+    "down.4.block.0.norm1.weight": "model-00002-of-00013.safetensors",
+    "down.4.block.0.norm2.bias": "model-00003-of-00013.safetensors",
+    "down.4.block.0.norm2.weight": "model-00003-of-00013.safetensors",
+    "down.4.block.0.temb_proj.bias": "model-00003-of-00013.safetensors",
+    "down.4.block.0.temb_proj.weight": "model-00003-of-00013.safetensors",
+    "down.4.block.1.conv1.bias": "model-00003-of-00013.safetensors",
+    "down.4.block.1.conv1.weight": "model-00003-of-00013.safetensors",
+    "down.4.block.1.conv2.bias": "model-00004-of-00013.safetensors",
+    "down.4.block.1.conv2.weight": "model-00004-of-00013.safetensors",
+    "down.4.block.1.norm1.bias": "model-00003-of-00013.safetensors",
+    "down.4.block.1.norm1.weight": "model-00003-of-00013.safetensors",
+    "down.4.block.1.norm2.bias": "model-00003-of-00013.safetensors",
+    "down.4.block.1.norm2.weight": "model-00003-of-00013.safetensors",
+    "down.4.block.1.temb_proj.bias": "model-00003-of-00013.safetensors",
+    "down.4.block.1.temb_proj.weight": "model-00003-of-00013.safetensors",
+    "mid.attn_1.k.bias": "model-00004-of-00013.safetensors",
+    "mid.attn_1.k.weight": "model-00004-of-00013.safetensors",
+    "mid.attn_1.norm.bias": "model-00004-of-00013.safetensors",
+    "mid.attn_1.norm.weight": "model-00004-of-00013.safetensors",
+    "mid.attn_1.proj_out.bias": "model-00004-of-00013.safetensors",
+    "mid.attn_1.proj_out.weight": "model-00004-of-00013.safetensors",
+    "mid.attn_1.q.bias": "model-00004-of-00013.safetensors",
+    "mid.attn_1.q.weight": "model-00004-of-00013.safetensors",
+    "mid.attn_1.v.bias": "model-00004-of-00013.safetensors",
+    "mid.attn_1.v.weight": "model-00004-of-00013.safetensors",
+    "mid.block_1.conv1.bias": "model-00004-of-00013.safetensors",
+    "mid.block_1.conv1.weight": "model-00004-of-00013.safetensors",
+    "mid.block_1.conv2.bias": "model-00004-of-00013.safetensors",
+    "mid.block_1.conv2.weight": "model-00004-of-00013.safetensors",
+    "mid.block_1.norm1.bias": "model-00004-of-00013.safetensors",
+    "mid.block_1.norm1.weight": "model-00004-of-00013.safetensors",
+    "mid.block_1.norm2.bias": "model-00004-of-00013.safetensors",
+    "mid.block_1.norm2.weight": "model-00004-of-00013.safetensors",
+    "mid.block_1.temb_proj.bias": "model-00004-of-00013.safetensors",
+    "mid.block_1.temb_proj.weight": "model-00004-of-00013.safetensors",
+    "mid.block_2.conv1.bias": "model-00005-of-00013.safetensors",
+    "mid.block_2.conv1.weight": "model-00005-of-00013.safetensors",
+    "mid.block_2.conv2.bias": "model-00005-of-00013.safetensors",
+    "mid.block_2.conv2.weight": "model-00005-of-00013.safetensors",
+    "mid.block_2.norm1.bias": "model-00004-of-00013.safetensors",
+    "mid.block_2.norm1.weight": "model-00004-of-00013.safetensors",
+    "mid.block_2.norm2.bias": "model-00005-of-00013.safetensors",
+    "mid.block_2.norm2.weight": "model-00005-of-00013.safetensors",
+    "mid.block_2.temb_proj.bias": "model-00005-of-00013.safetensors",
+    "mid.block_2.temb_proj.weight": "model-00005-of-00013.safetensors",
+    "norm_out.bias": "model-00013-of-00013.safetensors",
+    "norm_out.weight": "model-00013-of-00013.safetensors",
+    "temb.0.bias": "model-00001-of-00013.safetensors",
+    "temb.0.weight": "model-00001-of-00013.safetensors",
+    "temb.1.bias": "model-00001-of-00013.safetensors",
+    "temb.1.weight": "model-00001-of-00013.safetensors",
+    "up.0.attn.0.k.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.k.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.norm.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.norm.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.proj_out.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.proj_out.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.q.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.q.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.v.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.0.v.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.k.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.k.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.norm.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.norm.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.proj_out.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.proj_out.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.q.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.q.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.v.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.1.v.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.2.k.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.2.k.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.2.norm.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.2.norm.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.2.proj_out.bias": "model-00009-of-00013.safetensors",
+    "up.0.attn.2.proj_out.weight": "model-00009-of-00013.safetensors",
+    "up.0.attn.2.q.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.2.q.weight": "model-00008-of-00013.safetensors",
+    "up.0.attn.2.v.bias": "model-00008-of-00013.safetensors",
+    "up.0.attn.2.v.weight": "model-00008-of-00013.safetensors",
+    "up.0.block.0.conv1.bias": "model-00006-of-00013.safetensors",
+    "up.0.block.0.conv1.weight": "model-00006-of-00013.safetensors",
+    "up.0.block.0.conv2.bias": "model-00006-of-00013.safetensors",
+    "up.0.block.0.conv2.weight": "model-00006-of-00013.safetensors",
+    "up.0.block.0.norm1.bias": "model-00005-of-00013.safetensors",
+    "up.0.block.0.norm1.weight": "model-00005-of-00013.safetensors",
+    "up.0.block.0.norm2.bias": "model-00006-of-00013.safetensors",
+    "up.0.block.0.norm2.weight": "model-00006-of-00013.safetensors",
+    "up.0.block.0.shortcut.bias": "model-00006-of-00013.safetensors",
+    "up.0.block.0.shortcut.weight": "model-00006-of-00013.safetensors",
+    "up.0.block.0.temb_proj.bias": "model-00006-of-00013.safetensors",
+    "up.0.block.0.temb_proj.weight": "model-00006-of-00013.safetensors",
+    "up.0.block.1.conv1.bias": "model-00007-of-00013.safetensors",
+    "up.0.block.1.conv1.weight": "model-00007-of-00013.safetensors",
+    "up.0.block.1.conv2.bias": "model-00007-of-00013.safetensors",
+    "up.0.block.1.conv2.weight": "model-00007-of-00013.safetensors",
+    "up.0.block.1.norm1.bias": "model-00006-of-00013.safetensors",
+    "up.0.block.1.norm1.weight": "model-00006-of-00013.safetensors",
+    "up.0.block.1.norm2.bias": "model-00007-of-00013.safetensors",
+    "up.0.block.1.norm2.weight": "model-00007-of-00013.safetensors",
+    "up.0.block.1.shortcut.bias": "model-00007-of-00013.safetensors",
+    "up.0.block.1.shortcut.weight": "model-00007-of-00013.safetensors",
+    "up.0.block.1.temb_proj.bias": "model-00007-of-00013.safetensors",
+    "up.0.block.1.temb_proj.weight": "model-00007-of-00013.safetensors",
+    "up.0.block.2.conv1.bias": "model-00008-of-00013.safetensors",
+    "up.0.block.2.conv1.weight": "model-00008-of-00013.safetensors",
+    "up.0.block.2.conv2.bias": "model-00008-of-00013.safetensors",
+    "up.0.block.2.conv2.weight": "model-00008-of-00013.safetensors",
+    "up.0.block.2.norm1.bias": "model-00007-of-00013.safetensors",
+    "up.0.block.2.norm1.weight": "model-00007-of-00013.safetensors",
+    "up.0.block.2.norm2.bias": "model-00008-of-00013.safetensors",
+    "up.0.block.2.norm2.weight": "model-00008-of-00013.safetensors",
+    "up.0.block.2.shortcut.bias": "model-00008-of-00013.safetensors",
+    "up.0.block.2.shortcut.weight": "model-00008-of-00013.safetensors",
+    "up.0.block.2.temb_proj.bias": "model-00008-of-00013.safetensors",
+    "up.0.block.2.temb_proj.weight": "model-00008-of-00013.safetensors",
+    "up.0.upsample.conv.bias": "model-00009-of-00013.safetensors",
+    "up.0.upsample.conv.weight": "model-00009-of-00013.safetensors",
+    "up.1.attn.0.k.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.k.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.norm.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.norm.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.proj_out.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.proj_out.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.q.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.q.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.v.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.0.v.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.k.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.k.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.norm.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.norm.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.proj_out.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.proj_out.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.q.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.q.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.v.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.1.v.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.k.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.k.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.norm.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.norm.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.proj_out.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.proj_out.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.q.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.q.weight": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.v.bias": "model-00012-of-00013.safetensors",
+    "up.1.attn.2.v.weight": "model-00012-of-00013.safetensors",
+    "up.1.block.0.conv1.bias": "model-00009-of-00013.safetensors",
+    "up.1.block.0.conv1.weight": "model-00009-of-00013.safetensors",
+    "up.1.block.0.conv2.bias": "model-00010-of-00013.safetensors",
+    "up.1.block.0.conv2.weight": "model-00010-of-00013.safetensors",
+    "up.1.block.0.norm1.bias": "model-00009-of-00013.safetensors",
+    "up.1.block.0.norm1.weight": "model-00009-of-00013.safetensors",
+    "up.1.block.0.norm2.bias": "model-00009-of-00013.safetensors",
+    "up.1.block.0.norm2.weight": "model-00009-of-00013.safetensors",
+    "up.1.block.0.shortcut.bias": "model-00010-of-00013.safetensors",
+    "up.1.block.0.shortcut.weight": "model-00010-of-00013.safetensors",
+    "up.1.block.0.temb_proj.bias": "model-00009-of-00013.safetensors",
+    "up.1.block.0.temb_proj.weight": "model-00009-of-00013.safetensors",
+    "up.1.block.1.conv1.bias": "model-00010-of-00013.safetensors",
+    "up.1.block.1.conv1.weight": "model-00010-of-00013.safetensors",
+    "up.1.block.1.conv2.bias": "model-00011-of-00013.safetensors",
+    "up.1.block.1.conv2.weight": "model-00011-of-00013.safetensors",
+    "up.1.block.1.norm1.bias": "model-00010-of-00013.safetensors",
+    "up.1.block.1.norm1.weight": "model-00010-of-00013.safetensors",
+    "up.1.block.1.norm2.bias": "model-00010-of-00013.safetensors",
+    "up.1.block.1.norm2.weight": "model-00010-of-00013.safetensors",
+    "up.1.block.1.shortcut.bias": "model-00011-of-00013.safetensors",
+    "up.1.block.1.shortcut.weight": "model-00011-of-00013.safetensors",
+    "up.1.block.1.temb_proj.bias": "model-00010-of-00013.safetensors",
+    "up.1.block.1.temb_proj.weight": "model-00010-of-00013.safetensors",
+    "up.1.block.2.conv1.bias": "model-00011-of-00013.safetensors",
+    "up.1.block.2.conv1.weight": "model-00011-of-00013.safetensors",
+    "up.1.block.2.conv2.bias": "model-00012-of-00013.safetensors",
+    "up.1.block.2.conv2.weight": "model-00012-of-00013.safetensors",
+    "up.1.block.2.norm1.bias": "model-00011-of-00013.safetensors",
+    "up.1.block.2.norm1.weight": "model-00011-of-00013.safetensors",
+    "up.1.block.2.norm2.bias": "model-00011-of-00013.safetensors",
+    "up.1.block.2.norm2.weight": "model-00011-of-00013.safetensors",
+    "up.1.block.2.shortcut.bias": "model-00012-of-00013.safetensors",
+    "up.1.block.2.shortcut.weight": "model-00012-of-00013.safetensors",
+    "up.1.block.2.temb_proj.bias": "model-00011-of-00013.safetensors",
+    "up.1.block.2.temb_proj.weight": "model-00011-of-00013.safetensors",
+    "up.1.upsample.conv.bias": "model-00012-of-00013.safetensors",
+    "up.1.upsample.conv.weight": "model-00012-of-00013.safetensors",
+    "up.2.attn.0.k.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.k.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.norm.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.norm.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.proj_out.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.proj_out.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.q.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.q.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.v.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.0.v.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.k.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.k.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.norm.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.norm.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.proj_out.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.proj_out.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.q.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.q.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.v.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.1.v.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.k.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.k.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.norm.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.norm.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.proj_out.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.proj_out.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.q.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.q.weight": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.v.bias": "model-00013-of-00013.safetensors",
+    "up.2.attn.2.v.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.0.conv1.bias": "model-00012-of-00013.safetensors",
+    "up.2.block.0.conv1.weight": "model-00012-of-00013.safetensors",
+    "up.2.block.0.conv2.bias": "model-00012-of-00013.safetensors",
+    "up.2.block.0.conv2.weight": "model-00012-of-00013.safetensors",
+    "up.2.block.0.norm1.bias": "model-00012-of-00013.safetensors",
+    "up.2.block.0.norm1.weight": "model-00012-of-00013.safetensors",
+    "up.2.block.0.norm2.bias": "model-00012-of-00013.safetensors",
+    "up.2.block.0.norm2.weight": "model-00012-of-00013.safetensors",
+    "up.2.block.0.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.0.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.0.temb_proj.bias": "model-00012-of-00013.safetensors",
+    "up.2.block.0.temb_proj.weight": "model-00012-of-00013.safetensors",
+    "up.2.block.1.conv1.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.1.conv1.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.1.conv2.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.1.conv2.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.1.norm1.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.1.norm1.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.1.norm2.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.1.norm2.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.1.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.1.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.1.temb_proj.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.1.temb_proj.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.2.conv1.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.2.conv1.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.2.conv2.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.2.conv2.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.2.norm1.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.2.norm1.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.2.norm2.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.2.norm2.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.2.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.2.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.2.block.2.temb_proj.bias": "model-00013-of-00013.safetensors",
+    "up.2.block.2.temb_proj.weight": "model-00013-of-00013.safetensors",
+    "up.2.upsample.conv.bias": "model-00013-of-00013.safetensors",
+    "up.2.upsample.conv.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.0.conv1.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.0.conv1.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.0.conv2.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.0.conv2.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.0.norm1.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.0.norm1.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.0.norm2.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.0.norm2.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.0.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.0.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.0.temb_proj.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.0.temb_proj.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.1.conv1.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.1.conv1.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.1.conv2.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.1.conv2.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.1.norm1.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.1.norm1.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.1.norm2.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.1.norm2.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.1.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.1.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.1.temb_proj.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.1.temb_proj.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.2.conv1.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.2.conv1.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.2.conv2.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.2.conv2.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.2.norm1.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.2.norm1.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.2.norm2.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.2.norm2.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.2.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.2.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.3.block.2.temb_proj.bias": "model-00013-of-00013.safetensors",
+    "up.3.block.2.temb_proj.weight": "model-00013-of-00013.safetensors",
+    "up.3.upsample.conv.bias": "model-00013-of-00013.safetensors",
+    "up.3.upsample.conv.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.0.conv1.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.0.conv1.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.0.conv2.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.0.conv2.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.0.norm1.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.0.norm1.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.0.norm2.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.0.norm2.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.0.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.0.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.0.temb_proj.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.0.temb_proj.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.1.conv1.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.1.conv1.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.1.conv2.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.1.conv2.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.1.norm1.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.1.norm1.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.1.norm2.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.1.norm2.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.1.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.1.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.1.temb_proj.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.1.temb_proj.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.2.conv1.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.2.conv1.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.2.conv2.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.2.conv2.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.2.norm1.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.2.norm1.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.2.norm2.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.2.norm2.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.2.shortcut.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.2.shortcut.weight": "model-00013-of-00013.safetensors",
+    "up.4.block.2.temb_proj.bias": "model-00013-of-00013.safetensors",
+    "up.4.block.2.temb_proj.weight": "model-00013-of-00013.safetensors"
+  }
+}