End of training

.gitignore

@@ -1,4 +1,2 @@
 step_*
 epoch_*
-checkpoint-*
-

logs/text2image-fine-tune/1680507140.8057463/events.out.tfevents.1680507140.07872ee260ad.3270.1

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

logs/text2image-fine-tune/1680507140.8072975/hparams.yml

@@ -0,0 +1,47 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+cache_dir: /workspace/huggingface/cache
+caption_column: text
+center_crop: true
+checkpointing_steps: 5000
+checkpoints_total_limit: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: hearmeneigh/e621-rising-v2-curated
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: false
+hub_model_id: hearmeneigh/sd21-e621-rising-v2
+hub_token: null
+image_column: image
+learning_rate: 4.0e-06
+local_rank: 0
+logging_dir: logs
+lr_scheduler: constant
+lr_warmup_steps: 0
+maintain_aspect_ratio: true
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 1400
+mixed_precision: bf16
+noise_offset: 0
+non_ema_revision: null
+num_train_epochs: 1
+output_dir: /workspace/output/sd2-hmn-v2-step-4
+pretrained_model_name_or_path: /workspace/output/sd2-hmn-v2-step-3
+push_to_hub: true
+random_flip: true
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: null
+shuffle_tags: true
+train_batch_size: 34
+train_data_dir: null
+use_8bit_adam: false
+use_ema: true

logs/text2image-fine-tune/events.out.tfevents.1680507140.07872ee260ad.3270.0

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

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/workspace/output/sd2-hmn-v2-step-2",
+  "_name_or_path": "/workspace/output/sd2-hmn-v2-step-3",
   "architectures": [
     "CLIPTextModel"
   ],

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.14.0",
-  "_name_or_path": "/workspace/output/sd2-hmn-v2-step-3/checkpoint-1000",
+  "_name_or_path": "/workspace/output/sd2-hmn-v2-step-3",
   "act_fn": "silu",
   "attention_head_dim": [
     5,

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:87df050c333a81e18039cbecf1188e4572bcbf20ac9c24b39b1bc24e454dd6b3
+oid sha256:ea3a7fd5c062fe08d6a7253e5b3064e20278a0e1ed7daeb70487676067db32e4
 size 3463934693

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.14.0",
-  "_name_or_path": "/workspace/output/sd2-hmn-v2-step-2",
+  "_name_or_path": "/workspace/output/sd2-hmn-v2-step-3",
   "act_fn": "silu",
   "block_out_channels": [
     128,