diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4e6c55c624ebec7b1049de701b93088aa3865dc7 --- /dev/null +++ b/checkpoint-1000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:98371f777ddc28837debf3e4223a4e41f480a6b9a4b4418be043c3974db50a6e +size 4067517198 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..ee51fe3212c356acd53285517fee5536804188da --- /dev/null +++ b/checkpoint-1000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6bca4d15eeec17c49955b63371e1a8a4828783575ac6c101eecf225545b75782 +size 14408 diff --git a/checkpoint-1000/scaler.pt b/checkpoint-1000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..4562db25d2a589af962afdf91bdb8d84e80ba800 --- /dev/null +++ b/checkpoint-1000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:655ae68111f681abfb2cbb075959819e15773ccc65493f41a111ae5232129601 +size 988 diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..d0355bbef6ddedfa95ecbd5fe0a51c87171d6175 --- /dev/null +++ b/checkpoint-1000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:29ed599b4fc215d6c03af4802ccd7f2b52ff1a4a10b26acf08f5cbf68d4fd04a +size 1000 diff --git a/checkpoint-1000/transformer/config.json b/checkpoint-1000/transformer/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bde5a7ec6afbab2ba93735e0198219419ee5942a --- /dev/null +++ b/checkpoint-1000/transformer/config.json @@ -0,0 +1,16 @@ +{ + "_class_name": "SD3Transformer2DModel", + "_diffusers_version": "0.30.0.dev0", + "_name_or_path": "stabilityai/stable-diffusion-3-medium-diffusers", + "attention_head_dim": 64, + "caption_projection_dim": 1536, + "in_channels": 16, + "joint_attention_dim": 4096, + "num_attention_heads": 24, + "num_layers": 24, + "out_channels": 16, + "patch_size": 2, + "pooled_projection_dim": 2048, + "pos_embed_max_size": 192, + "sample_size": 128 +} diff --git a/checkpoint-1000/transformer/diffusion_pytorch_model.safetensors b/checkpoint-1000/transformer/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e1b28ddceafc7285d5dd08997127460670bbf853 --- /dev/null +++ b/checkpoint-1000/transformer/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4272962d2b969f3e7da45c400b7ba277e1d79ff95c6ebc5ed83a1b17f965f673 +size 8339883776 diff --git a/checkpoint-1500/optimizer.bin b/checkpoint-1500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..950339da7fc981d47485a5b00718e1cf007df0e5 --- /dev/null +++ b/checkpoint-1500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ca4ba4a9ab87ebfdd090e8fffadc7744bd48504c8d0118f9717f3b8dd17a9cf3 +size 4067517198 diff --git a/checkpoint-1500/random_states_0.pkl b/checkpoint-1500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..390c2febade1bcd953f828630db34c402846a4dd --- /dev/null +++ b/checkpoint-1500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7833e97d24e2656d51318a64cdf4e359c93c72a1b17f87c56e80010b6dc935d1 +size 14408 diff --git a/checkpoint-1500/scaler.pt b/checkpoint-1500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b8c640b8f9c0e74dc9f00f3c801dd7ece72843d9 --- /dev/null +++ b/checkpoint-1500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b92ef7781addb19e2d4dfa92f3ba360a4efafe712431eee5b78e639cbc25b404 +size 988 diff --git a/checkpoint-1500/scheduler.bin b/checkpoint-1500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..4765e9d039ea8d649082dda05a957047e3bc06ad --- /dev/null +++ b/checkpoint-1500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6d4c814ce7317c49b6f75baac8b2bbbb250473020f5c832a94b35aab52fa5bd9 +size 1000 diff --git a/checkpoint-1500/transformer/config.json b/checkpoint-1500/transformer/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bde5a7ec6afbab2ba93735e0198219419ee5942a --- /dev/null +++ b/checkpoint-1500/transformer/config.json @@ -0,0 +1,16 @@ +{ + "_class_name": "SD3Transformer2DModel", + "_diffusers_version": "0.30.0.dev0", + "_name_or_path": "stabilityai/stable-diffusion-3-medium-diffusers", + "attention_head_dim": 64, + "caption_projection_dim": 1536, + "in_channels": 16, + "joint_attention_dim": 4096, + "num_attention_heads": 24, + "num_layers": 24, + "out_channels": 16, + "patch_size": 2, + "pooled_projection_dim": 2048, + "pos_embed_max_size": 192, + "sample_size": 128 +} diff --git a/checkpoint-1500/transformer/diffusion_pytorch_model.safetensors b/checkpoint-1500/transformer/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..1099b4303c84f8211206b0b5e39010567e1815b5 --- /dev/null +++ b/checkpoint-1500/transformer/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8f5cc274b1f91f82a4c368c8787d6dd31d37f829646512b25e07888d978601f8 +size 8339883776 diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8b7bd09870608b0f02ffc265b95d9ebda441f3da --- /dev/null +++ b/checkpoint-2000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d2acbe5396ea0f3b8a023c3ef990234457bd69b009f08374470900d02e45cccd +size 4067517198 diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4efee2f9ee2f83c9632c5739c6f9122c2b521909 --- /dev/null +++ b/checkpoint-2000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a2899308b21ecfd4fe383cccad3d100f57856c5f69111e3024b9d9803f53b490 +size 14408 diff --git a/checkpoint-2000/scaler.pt b/checkpoint-2000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..6f4837110a8e86461445aecd7ed0cf8d2be79e24 --- /dev/null +++ b/checkpoint-2000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ad8dfb0c289e1642b1c3897010e9b30812501984c36731c1db27d14e904dadc4 +size 988 diff --git a/checkpoint-2000/scheduler.bin b/checkpoint-2000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..17da8e7652ee328d5dca077cbf901b9d32fa22f3 --- /dev/null +++ b/checkpoint-2000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:190299bfa26382211ed712ba8137894ab6b52151de9d20708e190793f4bf99ec +size 1000 diff --git a/checkpoint-2000/transformer/config.json b/checkpoint-2000/transformer/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bde5a7ec6afbab2ba93735e0198219419ee5942a --- /dev/null +++ b/checkpoint-2000/transformer/config.json @@ -0,0 +1,16 @@ +{ + "_class_name": "SD3Transformer2DModel", + "_diffusers_version": "0.30.0.dev0", + "_name_or_path": "stabilityai/stable-diffusion-3-medium-diffusers", + "attention_head_dim": 64, + "caption_projection_dim": 1536, + "in_channels": 16, + "joint_attention_dim": 4096, + "num_attention_heads": 24, + "num_layers": 24, + "out_channels": 16, + "patch_size": 2, + "pooled_projection_dim": 2048, + "pos_embed_max_size": 192, + "sample_size": 128 +} diff --git a/checkpoint-2000/transformer/diffusion_pytorch_model.safetensors b/checkpoint-2000/transformer/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..695787bae28360fc4e73bf1541fcfae67935550a --- /dev/null +++ b/checkpoint-2000/transformer/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6498091c5ab0b28c89d2cf729c53d27fdd104a7326812044fd4703b3f592fd8b +size 8339883776 diff --git a/checkpoint-2500/optimizer.bin b/checkpoint-2500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..96e9ebbd8a7de7c371b52451ff99abac4b6f2f81 --- /dev/null +++ b/checkpoint-2500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0227a720d5d777104680fb99960dcafe5f4a3792421b4c0bbf2e655e711f8b64 +size 4067517198 diff --git a/checkpoint-2500/random_states_0.pkl b/checkpoint-2500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..c42d2ffe1e6dc19642b3208c016ba5ee9a18718e --- /dev/null +++ b/checkpoint-2500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5b9c7a41096fadccc00dd38ae7c825de6b19e43ffcee01db2ad42fa1ca079f75 +size 14408 diff --git a/checkpoint-2500/scaler.pt b/checkpoint-2500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..0c690d0abb93bc87b55e366aafff18574f7fa159 --- /dev/null +++ b/checkpoint-2500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3cc13f9c286a0a85aca6431e94e59b51fcf0f32b8ae2b4c01dd64676a8b2c6f1 +size 988 diff --git a/checkpoint-2500/scheduler.bin b/checkpoint-2500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..e708599336d0bab1c46b1bc8b21d9f5cbf32f4f2 --- /dev/null +++ b/checkpoint-2500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8307bbf63fcc497e4cc4fc8fafa16037d67b7d15835cd24f8b6cc9b7d5939331 +size 1000 diff --git a/checkpoint-2500/transformer/config.json b/checkpoint-2500/transformer/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bde5a7ec6afbab2ba93735e0198219419ee5942a --- /dev/null +++ b/checkpoint-2500/transformer/config.json @@ -0,0 +1,16 @@ +{ + "_class_name": "SD3Transformer2DModel", + "_diffusers_version": "0.30.0.dev0", + "_name_or_path": "stabilityai/stable-diffusion-3-medium-diffusers", + "attention_head_dim": 64, + "caption_projection_dim": 1536, + "in_channels": 16, + "joint_attention_dim": 4096, + "num_attention_heads": 24, + "num_layers": 24, + "out_channels": 16, + "patch_size": 2, + "pooled_projection_dim": 2048, + "pos_embed_max_size": 192, + "sample_size": 128 +} diff --git a/checkpoint-2500/transformer/diffusion_pytorch_model.safetensors b/checkpoint-2500/transformer/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..8ff8a99c60530a148aa8b8da5a95109aa659cd90 --- /dev/null +++ b/checkpoint-2500/transformer/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a8033bc41acfce18d83ca8fa0cf428dd82fabb6d7cd375abed26296841de0ca0 +size 8339883776 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..45d27932e75f3a7591527c39316a0dfa6c2528f4 --- /dev/null +++ b/checkpoint-3000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:56877157e34b27c7bd14e332fa6a2b5eda923c7d49bc5c7784708d4ea9d92eef +size 4067517198 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..2090c77351972718ae2e57650b50ba43bdc456b7 --- /dev/null +++ b/checkpoint-3000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d739c37da1e6f4f8b5e273589398e7a87a827e8bc2f56b7f2524fea7d5edd80d +size 14408 diff --git a/checkpoint-3000/scaler.pt b/checkpoint-3000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..bfbb7a41487a74baed3d3f29daf45ed948f452d1 --- /dev/null +++ b/checkpoint-3000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d0613cd68c8e9f8cbdb521f917793035b18cce686834f07701e75a64ac00c960 +size 988 diff --git a/checkpoint-3000/scheduler.bin b/checkpoint-3000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..425a4144ba7feebfbbe75840bc8b424db7400c27 --- /dev/null +++ b/checkpoint-3000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a5e7a5426261d06dcc01a03c2e48fe5a4740c6195482a54bb267f6ff81207c5c +size 1000 diff --git a/checkpoint-3000/transformer/config.json b/checkpoint-3000/transformer/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bde5a7ec6afbab2ba93735e0198219419ee5942a --- /dev/null +++ b/checkpoint-3000/transformer/config.json @@ -0,0 +1,16 @@ +{ + "_class_name": "SD3Transformer2DModel", + "_diffusers_version": "0.30.0.dev0", + "_name_or_path": "stabilityai/stable-diffusion-3-medium-diffusers", + "attention_head_dim": 64, + "caption_projection_dim": 1536, + "in_channels": 16, + "joint_attention_dim": 4096, + "num_attention_heads": 24, + "num_layers": 24, + "out_channels": 16, + "patch_size": 2, + "pooled_projection_dim": 2048, + "pos_embed_max_size": 192, + "sample_size": 128 +} diff --git a/checkpoint-3000/transformer/diffusion_pytorch_model.safetensors b/checkpoint-3000/transformer/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..761d6d3a656a242fbce1daa470cb579278326b11 --- /dev/null +++ b/checkpoint-3000/transformer/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e4cd12ca0f4d2e0186fcb3a078dfd2c27c0587b3b41201d971e608435073e227 +size 8339883776 diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..e4eaf5d162995c61c341baeacd7c5e3ccb013daf --- /dev/null +++ b/checkpoint-3500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3739636dd8629774ed5c78a094ed91b6c189611b2492daf06b0beea53cffbe18 +size 4067517198 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..30dda36349b279d57ed75a74f01f5dc7c36e5257 --- /dev/null +++ b/checkpoint-3500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b25d0448c0bc2de70b743a98f77896fe87b12d1905cf85f281ed7d796f92210c +size 14408 diff --git a/checkpoint-3500/scaler.pt b/checkpoint-3500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..14bd1891e2ee1c8b2cbe2797e0751fcfd73040c3 --- /dev/null +++ b/checkpoint-3500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9d5147c6ee0c383b44fb6733bbdd74a265844e65ae3f124aeb6b51bc01687bb5 +size 988 diff --git a/checkpoint-3500/scheduler.bin b/checkpoint-3500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ffb934a137ebc8600138fc508e21217886654691 --- /dev/null +++ b/checkpoint-3500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:de1352ce06a177703ed735889d1587d0fcf4304955a9e3773225499fc1e8edfe +size 1000 diff --git a/checkpoint-3500/transformer/config.json b/checkpoint-3500/transformer/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bde5a7ec6afbab2ba93735e0198219419ee5942a --- /dev/null +++ b/checkpoint-3500/transformer/config.json @@ -0,0 +1,16 @@ +{ + "_class_name": "SD3Transformer2DModel", + "_diffusers_version": "0.30.0.dev0", + "_name_or_path": "stabilityai/stable-diffusion-3-medium-diffusers", + "attention_head_dim": 64, + "caption_projection_dim": 1536, + "in_channels": 16, + "joint_attention_dim": 4096, + "num_attention_heads": 24, + "num_layers": 24, + "out_channels": 16, + "patch_size": 2, + "pooled_projection_dim": 2048, + "pos_embed_max_size": 192, + "sample_size": 128 +} diff --git a/checkpoint-3500/transformer/diffusion_pytorch_model.safetensors b/checkpoint-3500/transformer/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a8bdd11427f99ee8e72758d4f019a1178d733da6 --- /dev/null +++ b/checkpoint-3500/transformer/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:83b5aca07cf3ac90b19fec1a8714cd04deaf061beda7fa7f269ce89547e8763e +size 8339883776 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f08d0f7112d9d367699a5c86d4bbbc49ea944ab6 --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7387f9dd344c6a7ab63d11a0d372bdcdd81a643fbb85130cf008517874466c0a +size 4067517198 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..dbed0bfff99b6e84e81d26fecff65c77b6be4a10 --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e34225a16c79a131629e6f5d6b2c435f96e562021669f0711ad0c1be0ea1e5d1 +size 14408 diff --git a/checkpoint-4000/scaler.pt b/checkpoint-4000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..7defe84db6c9b72ddc31e68cc1610e5385d2c180 --- /dev/null +++ b/checkpoint-4000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1f710259cf7374ef56e02fe7142c2dd927e7b222cbe9df0da4a46ba0ae5ba385 +size 988 diff --git a/checkpoint-4000/scheduler.bin b/checkpoint-4000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..3c24d46de78f3fd3b3874bdab622244690194d76 --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ff519dccdfd88166ca4efa26accd0379d9a870ad3d44f934318830d4d730c17d +size 1000 diff --git a/checkpoint-4000/transformer/config.json b/checkpoint-4000/transformer/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bde5a7ec6afbab2ba93735e0198219419ee5942a --- /dev/null +++ b/checkpoint-4000/transformer/config.json @@ -0,0 +1,16 @@ +{ + "_class_name": "SD3Transformer2DModel", + "_diffusers_version": "0.30.0.dev0", + "_name_or_path": "stabilityai/stable-diffusion-3-medium-diffusers", + "attention_head_dim": 64, + "caption_projection_dim": 1536, + "in_channels": 16, + "joint_attention_dim": 4096, + "num_attention_heads": 24, + "num_layers": 24, + "out_channels": 16, + "patch_size": 2, + "pooled_projection_dim": 2048, + "pos_embed_max_size": 192, + "sample_size": 128 +} diff --git a/checkpoint-4000/transformer/diffusion_pytorch_model.safetensors b/checkpoint-4000/transformer/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a644882897f7f0a703b3dda2cf3bb4d7ba8c2a3c --- /dev/null +++ b/checkpoint-4000/transformer/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3a381b564be64bd51365bd4ffef772ffaabbb02e8b62317a0c3af31f2f1b5746 +size 8339883776 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin index f0799f14c8525653473df0c9a0fb5837e420b157..eb6579499de4b0a5ffa5161002699baaee6d9308 100644 --- a/checkpoint-500/optimizer.bin +++ b/checkpoint-500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:ab49bc931fd5d171d64dc6eb273c09e3ec673c672ed3f6b5ce596e08c020e00f +oid sha256:93d4572c11627baf340a702c5de0ccd8a96fdfae765f1c4c33c8f7110cbc3db0 size 4067517198 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl index a11732cf3cb8073c8510be9094f615b0ac6caf96..8d47ee3f66be9cce0e1743b3e8aa9c490bb2583b 100644 --- a/checkpoint-500/random_states_0.pkl +++ b/checkpoint-500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:287c3776a9205e57700f17650245ad75278139469fe0ef68e95e6e0aaa2e8ec3 -size 14344 +oid sha256:40c219cb7cb52e7d344b6c3ae1b96b807d915d33a4adbeff15e25a7c9b029ee2 +size 14408 diff --git a/checkpoint-500/scaler.pt b/checkpoint-500/scaler.pt index b1640696e8c648247d0af4fea4d97302082e3749..e3581673b7c8508713b9f1cb2bc0a23ce50d7d5d 100644 --- a/checkpoint-500/scaler.pt +++ b/checkpoint-500/scaler.pt @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:18b984273ea2d45b7ffb1d047bb359d93111e41fcad70d16a1b453fd38f72636 +oid sha256:6698dfaf44eb77634b79ade57e1e7f475241038250c3a4893dba02411ec4be5d size 988 diff --git a/checkpoint-500/scheduler.bin b/checkpoint-500/scheduler.bin index 5f0c87cfc1ffdc1fa8e66ebda086067a4ae7088b..ebff488daf3705bca67cfe6e41b91856bd500552 100644 --- a/checkpoint-500/scheduler.bin +++ b/checkpoint-500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:6cf34ac8dd2addc454aeb1ee4eaba3267c2841857e117be40ac2ed02cf581c9c +oid sha256:e38e154212a25b27bdeeae83b49a20a33b1395fcb19d73d049d76a11463c13cb size 1000 diff --git a/checkpoint-500/transformer/diffusion_pytorch_model.safetensors b/checkpoint-500/transformer/diffusion_pytorch_model.safetensors index 35e4273425dae4b4dea241e2f147ec20993addc5..0eeb3c9b667dde09e63d6c52d0f6939f6aa4027d 100644 --- a/checkpoint-500/transformer/diffusion_pytorch_model.safetensors +++ b/checkpoint-500/transformer/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:17adf2446401c7fffb55e6a225e717583fba66e9ca2584f4b7d17c3db6f16c12 +oid sha256:91f2da5e38dc9672c8678867cb3eca624c59b530a91608575afeb900d38cbc30 size 8339883776 diff --git a/logs/dreambooth-sd3/1721694433.137214/events.out.tfevents.1721694433.fe554585b0d9.8107.1 b/logs/dreambooth-sd3/1721694433.137214/events.out.tfevents.1721694433.fe554585b0d9.8107.1 new file mode 100644 index 0000000000000000000000000000000000000000..3023ba464b17cb19b97daf5f2d637fba55cb7833 --- /dev/null +++ b/logs/dreambooth-sd3/1721694433.137214/events.out.tfevents.1721694433.fe554585b0d9.8107.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b53748209bfd9adbe40d6d35caec108000588272c0d6ff444a2f57e51ba4a424 +size 3058 diff --git a/logs/dreambooth-sd3/1721694433.1391208/hparams.yml b/logs/dreambooth-sd3/1721694433.1391208/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..0edb6f5514593425710d34f234d378c3ce11fec3 --- /dev/null +++ b/logs/dreambooth-sd3/1721694433.1391208/hparams.yml @@ -0,0 +1,68 @@ +adam_beta1: 0.9 +adam_beta2: 0.999 +adam_epsilon: 1.0e-08 +adam_weight_decay: 0.0001 +adam_weight_decay_text_encoder: 0.001 +allow_tf32: false +cache_dir: null +caption_column: null +center_crop: false +checkpointing_steps: 500 +checkpoints_total_limit: null +class_data_dir: null +class_prompt: null +dataloader_num_workers: 0 +dataset_config_name: null +dataset_name: null +gradient_accumulation_steps: 4 +gradient_checkpointing: false +hub_model_id: null +hub_token: null +image_column: image +instance_data_dir: dog +instance_prompt: a photo of sks dog +learning_rate: 1.0e-06 +local_rank: -1 +logging_dir: logs +logit_mean: 0.0 +logit_std: 1.0 +lr_num_cycles: 1 +lr_power: 1.0 +lr_scheduler: constant +lr_warmup_steps: 0 +max_grad_norm: 1.0 +max_sequence_length: 77 +max_train_steps: 4000 +mixed_precision: fp16 +mode_scale: 1.29 +num_class_images: 100 +num_train_epochs: 2000 +num_validation_images: 4 +optimizer: AdamW +output_dir: trained-sd3 +pretrained_model_name_or_path: stabilityai/stable-diffusion-3-medium-diffusers +prior_generation_precision: null +prior_loss_weight: 1.0 +prodigy_beta3: null +prodigy_decouple: true +prodigy_safeguard_warmup: true +prodigy_use_bias_correction: true +push_to_hub: true +random_flip: false +repeats: 1 +report_to: tensorboard +resolution: 1024 +resume_from_checkpoint: null +revision: null +sample_batch_size: 4 +scale_lr: false +seed: 1337 +text_encoder_lr: 5.0e-06 +train_batch_size: 1 +train_text_encoder: true +use_8bit_adam: true +validation_epochs: 25 +validation_prompt: null +variant: null +weighting_scheme: logit_normal +with_prior_preservation: false diff --git a/logs/dreambooth-sd3/1721694656.7422438/events.out.tfevents.1721694656.fe554585b0d9.9411.1 b/logs/dreambooth-sd3/1721694656.7422438/events.out.tfevents.1721694656.fe554585b0d9.9411.1 new file mode 100644 index 0000000000000000000000000000000000000000..dc893dbfad5f7c255ed3e0a8873554238fbbee14 --- /dev/null +++ b/logs/dreambooth-sd3/1721694656.7422438/events.out.tfevents.1721694656.fe554585b0d9.9411.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d8a20ae079f71ba4afb75e02837d19ba5560d2d72175ee925f0d181dae4d1d59 +size 3058 diff --git a/logs/dreambooth-sd3/1721694656.74416/hparams.yml b/logs/dreambooth-sd3/1721694656.74416/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..1aa3bb90f216b5f4db32ffc7bc0a2208173dc404 --- /dev/null +++ b/logs/dreambooth-sd3/1721694656.74416/hparams.yml @@ -0,0 +1,68 @@ +adam_beta1: 0.9 +adam_beta2: 0.999 +adam_epsilon: 1.0e-08 +adam_weight_decay: 0.0001 +adam_weight_decay_text_encoder: 0.001 +allow_tf32: false +cache_dir: null +caption_column: null +center_crop: false +checkpointing_steps: 500 +checkpoints_total_limit: null +class_data_dir: null +class_prompt: null +dataloader_num_workers: 0 +dataset_config_name: null +dataset_name: null +gradient_accumulation_steps: 4 +gradient_checkpointing: false +hub_model_id: null +hub_token: null +image_column: image +instance_data_dir: dog +instance_prompt: a photo of sks dog +learning_rate: 1.0e-06 +local_rank: -1 +logging_dir: logs +logit_mean: 0.0 +logit_std: 1.0 +lr_num_cycles: 1 +lr_power: 1.0 +lr_scheduler: constant +lr_warmup_steps: 0 +max_grad_norm: 1.0 +max_sequence_length: 77 +max_train_steps: 4000 +mixed_precision: fp16 +mode_scale: 1.29 +num_class_images: 100 +num_train_epochs: 2000 +num_validation_images: 4 +optimizer: AdamW +output_dir: trained-sd3 +pretrained_model_name_or_path: stabilityai/stable-diffusion-3-medium-diffusers +prior_generation_precision: null +prior_loss_weight: 1.0 +prodigy_beta3: null +prodigy_decouple: true +prodigy_safeguard_warmup: true +prodigy_use_bias_correction: true +push_to_hub: true +random_flip: false +repeats: 1 +report_to: tensorboard +resolution: 1024 +resume_from_checkpoint: null +revision: null +sample_batch_size: 4 +scale_lr: false +seed: 1337 +text_encoder_lr: 5.0e-06 +train_batch_size: 1 +train_text_encoder: false +use_8bit_adam: true +validation_epochs: 25 +validation_prompt: null +variant: null +weighting_scheme: logit_normal +with_prior_preservation: false diff --git a/logs/dreambooth-sd3/events.out.tfevents.1721694433.fe554585b0d9.8107.0 b/logs/dreambooth-sd3/events.out.tfevents.1721694433.fe554585b0d9.8107.0 new file mode 100644 index 0000000000000000000000000000000000000000..aafc18c56ef830cfed3b4d505ccec7d8d77805f7 --- /dev/null +++ b/logs/dreambooth-sd3/events.out.tfevents.1721694433.fe554585b0d9.8107.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2c737ab4ceb9c098b4efd11ccf617fa4dbdfc641c6a0e20f1ff20654baca4cd6 +size 88 diff --git a/logs/dreambooth-sd3/events.out.tfevents.1721694656.fe554585b0d9.9411.0 b/logs/dreambooth-sd3/events.out.tfevents.1721694656.fe554585b0d9.9411.0 new file mode 100644 index 0000000000000000000000000000000000000000..c7aefcfed159fa705c570fb0e4f98b70191259c8 --- /dev/null +++ b/logs/dreambooth-sd3/events.out.tfevents.1721694656.fe554585b0d9.9411.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:16f74e11ea9d4e7ffaac442c2fa88b4667bb65f94ad4159336167200d75a01a7 +size 839438 diff --git a/transformer/diffusion_pytorch_model.safetensors b/transformer/diffusion_pytorch_model.safetensors index 35e4273425dae4b4dea241e2f147ec20993addc5..a644882897f7f0a703b3dda2cf3bb4d7ba8c2a3c 100644 --- a/transformer/diffusion_pytorch_model.safetensors +++ b/transformer/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:17adf2446401c7fffb55e6a225e717583fba66e9ca2584f4b7d17c3db6f16c12 +oid sha256:3a381b564be64bd51365bd4ffef772ffaabbb02e8b62317a0c3af31f2f1b5746 size 8339883776 diff --git a/vae/config.json b/vae/config.json index f080183ddae1ad9bd7221a6c735a80ac1b7d733c..50425ebe2e95c4d97c52598954f594b273ae404b 100644 --- a/vae/config.json +++ b/vae/config.json @@ -21,6 +21,7 @@ "latents_mean": null, "latents_std": null, "layers_per_block": 2, + "mid_block_add_attention": true, "norm_num_groups": 32, "out_channels": 3, "sample_size": 1024,