End of training

README.md

@@ -1,7 +1,7 @@
 
 ---
 license: creativeml-openrail-m
-base_model: CompVis/stable-diffusion-v1-4
+base_model: SG161222/Realistic_Vision_V2.0
 instance_prompt: a photo of sks cat
 tags:
 - stable-diffusion
@@ -14,7 +14,7 @@ inference: true
     
 # DreamBooth - ainjarts/model
 
-This is a dreambooth model derived from CompVis/stable-diffusion-v1-4. The weights were trained on a photo of sks cat using [DreamBooth](https://dreambooth.github.io/).
+This is a dreambooth model derived from SG161222/Realistic_Vision_V2.0. The weights were trained on a photo of sks cat 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:fc3fe2074421de6a08e1bec7b5ac0d661ba5faf27c01e7a08bcdad42089c7ed4
+oid sha256:f68592290c9811bc3ed94edc6a27b51eab3cb72bc3caaae3fc6f3af40236440d
 size 1972598422

checkpoint-500/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f5479aa873838b3d3d05ffb30f0868457615ec177d53ef3fff8971bce81b0fa0
-size 14344
+oid sha256:60192ff37cadf3bf9f9e0e5e67bb1e3c725a0a49ecdc06115ab1e7b3f1b75543
+size 14408

checkpoint-500/text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "CompVis/stable-diffusion-v1-4",
+  "_name_or_path": "SG161222/Realistic_Vision_V2.0",
   "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": "float32",
   "transformers_version": "4.35.2",
   "vocab_size": 49408

checkpoint-500/text_encoder/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8d986f55e835d0246193d8715f9f541378b26187bba4c0c829c09ce065cd1a2e
+oid sha256:9fe3495586c166fb3c9a9832c69a80fc9472987cde2064d2c31b129142fc0bff
 size 492265168

checkpoint-500/unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.24.0.dev0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-4",
+  "_name_or_path": "SG161222/Realistic_Vision_V2.0",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

checkpoint-500/unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:408b0f52300b5167b0cce1950277de0710742a32a8e1b48d12f8d995b0cfd092
+oid sha256:3b8d35113e58cabb18a74d3ec4bde5bc558fc157823b40f70c220130eb982ac5
 size 3438167536

logs/dreambooth/1701169490.5830095/events.out.tfevents.1701169490.53ad7ae7f465.1631.1

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

logs/dreambooth/1701169490.585203/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: /home/diffusers/examples/dreambooth/dataset/img/class
+class_labels_conditioning: null
+class_prompt: a photo of cat
+dataloader_num_workers: 0
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+instance_data_dir: /home/diffusers/examples/dreambooth/dataset/img/cat
+instance_prompt: a photo of sks cat
+learning_rate: 2.0e-06
+local_rank: -1
+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: 800
+mixed_precision: null
+num_class_images: 50
+num_train_epochs: 16
+num_validation_images: 4
+offset_noise: false
+output_dir: /home/diffusers/examples/dreambooth/dataset/model
+pre_compute_text_embeddings: false
+pretrained_model_name_or_path: SG161222/Realistic_Vision_V2.0
+prior_generation_precision: null
+prior_loss_weight: 1.0
+push_to_hub: true
+report_to: tensorboard
+resolution: 1024
+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: true
+use_8bit_adam: true
+validation_prompt: null
+validation_scheduler: DPMSolverMultistepScheduler
+validation_steps: 100
+variant: null
+with_prior_preservation: true

logs/dreambooth/events.out.tfevents.1701169490.53ad7ae7f465.1631.0

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

model_index.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.24.0.dev0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-4",
+  "_name_or_path": "SG161222/Realistic_Vision_V2.0",
   "feature_extractor": [
     "transformers",
     "CLIPImageProcessor"

safety_checker/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/home/.cache/huggingface/hub/models--CompVis--stable-diffusion-v1-4/snapshots/133a221b8aa7292a167afc5127cb63fb5005638b/safety_checker",
+  "_name_or_path": "/home/.cache/huggingface/hub/models--SG161222--Realistic_Vision_V2.0/snapshots/60c57a68e17cc9261e6031cb3babd7ffd6d6dde1/safety_checker",
   "architectures": [
     "StableDiffusionSafetyChecker"
   ],

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "CompVis/stable-diffusion-v1-4",
+  "_name_or_path": "SG161222/Realistic_Vision_V2.0",
   "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": "float32",
   "transformers_version": "4.35.2",
   "vocab_size": 49408

text_encoder/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:074d84b424cfb7c708bba569a40bfde8700fe056f55d5c8a92ad8774c317023a
+oid sha256:6a169f538e06a896d3e311602e97ddac9c3b144a72bf7bb147bfd1a76bb5cdc3
 size 492265168

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.24.0.dev0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-4",
+  "_name_or_path": "SG161222/Realistic_Vision_V2.0",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:794df22d40c963ee0d3e7315a3a3bcc9bf43263a613359b3dd3809925a4f2ab6
+oid sha256:cabbf3804ebc108c3397ec4e961d1c9365b7391c9f35952bb09924c64fc39b22
 size 3438167536

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.24.0.dev0",
-  "_name_or_path": "/home/.cache/huggingface/hub/models--CompVis--stable-diffusion-v1-4/snapshots/133a221b8aa7292a167afc5127cb63fb5005638b/vae",
+  "_name_or_path": "/home/.cache/huggingface/hub/models--SG161222--Realistic_Vision_V2.0/snapshots/60c57a68e17cc9261e6031cb3babd7ffd6d6dde1/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,

vae/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b4d2b5932bb4151e54e694fd31ccf51fca908223c9485bd56cd0e1d83ad94c49
+oid sha256:86a6170968f284b6c0c111e2611c806a04102e2a48790ebd1d490ca78040d2bb
 size 334643268