End of training

README.md

@@ -2,7 +2,7 @@
 ---
 license: creativeml-openrail-m
 base_model: stabilityai/stable-diffusion-2-1
-instance_prompt: a photo of a gdm gundam
+instance_prompt: a photo of a sks gundam
 tags:
 - stable-diffusion
 - stable-diffusion-diffusers
@@ -14,7 +14,7 @@ inference: true
     
 # DreamBooth - briannlongzhao/gundam_dreambooth
 
-This is a dreambooth model derived from stabilityai/stable-diffusion-2-1. The weights were trained on a photo of a gdm gundam using [DreamBooth](https://dreambooth.github.io/).
+This is a dreambooth model derived from stabilityai/stable-diffusion-2-1. The weights were trained on a photo of a sks gundam using [DreamBooth](https://dreambooth.github.io/).
 You can find some example images in the following. 
 
 

checkpoint-500/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:30542f4aaca5c1cb5b5bb7fc7c85bc06d5fa7d22ca93d76e8545635b437855ea
-size 1737914938
+oid sha256:23f0ba8e2e49c09eaec4828717cd1e2d19ff55f3e1d2819c964cad9f2a51cbf4
+size 1737914501

checkpoint-500/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b34f26e8ba60539d9306396c552eec6205b6d2ecf38484dd603a02349de23802
-size 14408
+oid sha256:0a1b827b31b77c5b2371c35cd6e010c1457740f940f175337cbefbbdfbd51376
+size 15691

checkpoint-500/scaler.pt

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

checkpoint-500/scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:65115106fbd9463fa6c7133b543dece71c31fcba363ff15b8c73f8c7eb0739c5
-size 1000
+oid sha256:a1e6017f7bb591dfb5c0c7aebbd8ee312a2453b8874f27af36143675db9fc905
+size 563

checkpoint-500/unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:67cba892328fb04a452330eb4e40bf807e7142e5374af95df6bd2583c7563bb1
+oid sha256:16f50795890d7f409855305afbedffeb630273ba79f20183ef4e1d82caa3ce9e
 size 3463726504

logs/dreambooth/1732586533.807998/events.out.tfevents.1732586533.iGpu14.3234248.1

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

logs/dreambooth/1732586533.8157568/hparams.yml

@@ -0,0 +1,60 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+center_crop: false
+checkpointing_steps: 500
+checkpoints_total_limit: null
+class_data_dir: /lab/tmpig7b/u/brian-data/eval_4/gundam_preserve
+class_labels_conditioning: null
+class_prompt: a photo of a gundam
+dataloader_num_workers: 0
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+idx: null
+instance_data_dir: /lab/tmpig7b/u/brian-data/eval_4/gundam/reference
+instance_prompt: a photo of a sks gundam
+learning_rate: 5.0e-06
+local_rank: 0
+logging_dir: logs
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_steps: 500
+mixed_precision: fp16
+num_class_images: 50
+num_train_epochs: 20
+num_validation_images: 4
+offset_noise: false
+output_dir: dreambooth_models/gundam_dreambooth
+pre_compute_text_embeddings: false
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
+prior_generation_precision: null
+prior_loss_weight: 1.0
+push_to_hub: true
+report_to: tensorboard
+resolution: 768
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: null
+set_grads_to_none: false
+skip_save_text_encoder: false
+snr_gamma: null
+text_encoder_use_attention_mask: false
+tokenizer_max_length: null
+tokenizer_name: null
+train_batch_size: 1
+train_text_encoder: false
+use_8bit_adam: true
+validation_prompt: null
+validation_scheduler: DPMSolverMultistepScheduler
+validation_steps: 100
+with_prior_preservation: true

logs/dreambooth/events.out.tfevents.1732586533.iGpu14.3234248.0

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

text_encoder/config.json

@@ -19,7 +19,7 @@
   "num_hidden_layers": 23,
   "pad_token_id": 1,
   "projection_dim": 512,
-  "torch_dtype": "bfloat16",
-  "transformers_version": "4.34.1",
+  "torch_dtype": "float16",
+  "transformers_version": "4.28.1",
   "vocab_size": 49408
 }

text_encoder/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:305b018799a2e1dba83e9db17654309814ff4212e40b8096b237ce2e3794a132
-size 680820760
+oid sha256:681c555376658c81dc273f2d737a2aeb23ddb6d1d8e5b3a7064636d359a22668
+size 680821096

tokenizer/tokenizer_config.json

@@ -1,38 +1,33 @@
 {
   "add_prefix_space": false,
-  "added_tokens_decoder": {
-    "0": {
-      "content": "!",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "49406": {
-      "content": "<|startoftext|>",
-      "lstrip": false,
-      "normalized": true,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "49407": {
-      "content": "<|endoftext|>",
-      "lstrip": false,
-      "normalized": true,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    }
+  "bos_token": {
+    "__type": "AddedToken",
+    "content": "<|startoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
   },
-  "bos_token": "<|startoftext|>",
   "clean_up_tokenization_spaces": true,
   "do_lower_case": true,
-  "eos_token": "<|endoftext|>",
+  "eos_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
   "errors": "replace",
   "model_max_length": 77,
-  "pad_token": "!",
+  "pad_token": "<|endoftext|>",
   "tokenizer_class": "CLIPTokenizer",
-  "unk_token": "<|endoftext|>"
+  "unk_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
 }

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:045940e03e551784d4ba07c4ab796e40a956f07aac98bccc449fc3f9dc4f1037
+oid sha256:16f50795890d7f409855305afbedffeb630273ba79f20183ef4e1d82caa3ce9e
 size 3463726504

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.22.0.dev0",
-  "_name_or_path": "/home/v-enlongzhao/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6/vae",
+  "_name_or_path": "/lab/briannlz/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,