[retrained: based on ptx0/pseudo-journey @ 4000 steps from stable-diffusion-2-1 baseline on 3300 images] + 9500 steps on 22,400 images, polynomial learning rate scheduler, batch size 4, 64 gradient accumulations, FROZEN text encoder, 8bit ADAM, ZERO PLW (no regularization data), followed by 550 steps with unfrozen text encoder and constant LR 1e-8

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/notebooks/images/datasets/models//checkpoint-13000/text_encoder",
+  "_name_or_path": "/notebooks/images/datasets/models//checkpoint-550/text_encoder",
   "architectures": [
     "CLIPTextModel"
   ],

text_encoder/{pytorch_model.bin → model.safetensors}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:03c3477c3db2f7edb53d631293d9a7018c4b0a23cee490b60cb049aa9bd54e95
-size 1361679905
+oid sha256:d7f8471deacb5f976911bfcf000ec5ea27e912b491dfd5041d2bbeefe88ebf8a
+size 1361597016

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.17.0.dev0",
-  "_name_or_path": "/notebooks/images/datasets/models//checkpoint-13000/unet",
+  "_name_or_path": "/notebooks/images/datasets/models//checkpoint-550/unet",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

unet/{diffusion_pytorch_model.bin → diffusion_pytorch_model.safetensors}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6636f688759f6b44b8d332ec0a6fc513cba59c49240d4170f45b8579d2071e6f
-size 3463934693
+oid sha256:644a1dd448e2ac8faba304c89ae1c0f65e9d73ef84144b3d5d3ef22915b81f76
+size 3463726504

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.17.0.dev0",
-  "_name_or_path": "/root/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/845609e6cf0a060d8cd837297e5c169df5bff72c/vae",
+  "_name_or_path": "/notebooks/datasets/models/pipeline/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,

vae/{diffusion_pytorch_model.bin → diffusion_pytorch_model.safetensors}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:185b0c03485b4048bb6158de087df301b79ea187844c76ae91cd4cda207282a2
-size 334715569
+oid sha256:2aa1f43011b553a4cba7f37456465cdbd48aab7b54b9348b890e8058ea7683ec
+size 334643268