End of training

logs/text2image-fine-tune/1686346593.8511884/events.out.tfevents.1686346593.2d86f0c58c3a.22128.1

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

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

@@ -0,0 +1,50 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+cache_dir: null
+caption_column: text
+center_crop: true
+checkpointing_steps: 100000
+checkpoints_total_limit: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: mswhite/nagel_dataset
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+image_column: image
+input_perturbation: 0
+learning_rate: 1.0e-05
+local_rank: -1
+logging_dir: logs
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 400
+mixed_precision: fp16
+noise_offset: 0
+non_ema_revision: null
+num_train_epochs: 80
+output_dir: miamivice
+prediction_type: null
+pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
+push_to_hub: true
+random_flip: true
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: null
+snr_gamma: null
+tracker_project_name: text2image-fine-tune
+train_batch_size: 1
+train_data_dir: null
+use_8bit_adam: false
+use_ema: true
+validation_epochs: 5

logs/text2image-fine-tune/events.out.tfevents.1686346593.2d86f0c58c3a.22128.0

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

safety_checker/config.json

@@ -1,6 +1,6 @@
 {
-  "_commit_hash": "d23f1896b39c34425e36569b72bf037d36766fad",
-  "_name_or_path": "/root/.cache/huggingface/hub/models--CompVis--stable-diffusion-v1-2/snapshots/d23f1896b39c34425e36569b72bf037d36766fad/safety_checker",
+  "_commit_hash": "aa9ba505e1973ae5cd05f5aedd345178f52f8e6a",
+  "_name_or_path": "/root/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/aa9ba505e1973ae5cd05f5aedd345178f52f8e6a/safety_checker",
   "architectures": [
     "StableDiffusionSafetyChecker"
   ],

scheduler/scheduler_config.json

@@ -4,6 +4,7 @@
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,
+  "clip_sample": false,
   "num_train_timesteps": 1000,
   "prediction_type": "epsilon",
   "set_alpha_to_one": false,

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "CompVis/stable-diffusion-v1-2",
+  "_name_or_path": "runwayml/stable-diffusion-v1-5",
   "architectures": [
     "CLIPTextModel"
   ],
@@ -18,7 +18,7 @@
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 1,
-  "projection_dim": 512,
+  "projection_dim": 768,
   "torch_dtype": "float16",
   "transformers_version": "4.30.1",
   "vocab_size": 49408

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.18.0.dev0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-2",
+  "_name_or_path": "runwayml/stable-diffusion-v1-5",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:75c35af69a83626a49d5ec96e2924d34db3bfe01e3390406661a4443cb9d4509
+oid sha256:633f9a35d17174542a4261520e9e78600a1af663c628132045557d7941a28704
 size 3438375973

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.18.0.dev0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-2",
+  "_name_or_path": "runwayml/stable-diffusion-v1-5",
   "act_fn": "silu",
   "block_out_channels": [
     128,