End of training

Files changed (11) hide show

README.md CHANGED Viewed

@@ -10,6 +10,12 @@ tags:
 - diffusers
 - controlnet
 - diffusers-training
 ---
 <!-- This model card has been generated automatically according to the information the training script had access to. You

 - diffusers
 - controlnet
 - diffusers-training
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+- diffusers
+- controlnet
+- diffusers-training
 ---
 <!-- This model card has been generated automatically according to the information the training script had access to. You

config.json CHANGED Viewed

@@ -5,7 +5,12 @@
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,
   "addition_time_embed_dim": null,
-  "attention_head_dim": 8,
   "block_out_channels": [
     320,
     640,
@@ -21,7 +26,7 @@
     256
   ],
   "controlnet_conditioning_channel_order": "rgb",
-  "cross_attention_dim": 768,
   "down_block_types": [
     "CrossAttnDownBlock2D",
     "CrossAttnDownBlock2D",
@@ -46,6 +51,6 @@
   "projection_class_embeddings_input_dim": null,
   "resnet_time_scale_shift": "default",
   "transformer_layers_per_block": 1,
-  "upcast_attention": false,
-  "use_linear_projection": false
 }

   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,
   "addition_time_embed_dim": null,
+  "attention_head_dim": [
+    5,
+    10,
+    20,
+    20
+  ],
   "block_out_channels": [
     320,
     640,
     256
   ],
   "controlnet_conditioning_channel_order": "rgb",
+  "cross_attention_dim": 1024,
   "down_block_types": [
     "CrossAttnDownBlock2D",
     "CrossAttnDownBlock2D",
   "projection_class_embeddings_input_dim": null,
   "resnet_time_scale_shift": "default",
   "transformer_layers_per_block": 1,
+  "upcast_attention": true,
+  "use_linear_projection": true
 }

diffusion_pytorch_model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:36af13c8754c5e4c9c60de38a6911a9ce1b48c09014029aa6997862342f561de
+size 1456953560

image_control.png CHANGED Viewed

images_0.png CHANGED Viewed

logs/train_controlnet_persian/1757353931.5153167/events.out.tfevents.1757353931.0f16519829e8.4893.1 ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:45905fe4b32aea20c868a430bdc9152dd8e91dd45cd5c2f0ce975c29654d8ee4
+size 2355

logs/train_controlnet_persian/1757353931.5166864/hparams.yml ADDED Viewed

+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
+checkpointing_steps: 1000
+checkpoints_total_limit: null
+conditioning_image_column: conditioning_image
+controlnet_model_name_or_path: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: null
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+image_column: image
+learning_rate: 1.0e-05
+logging_dir: logs
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 500
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 90
+mixed_precision: fp16
+num_train_epochs: 15
+num_validation_images: 4
+output_dir: ./controlnet_persian_text_v1
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
+proportion_empty_prompts: 0
+push_to_hub: true
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: null
+set_grads_to_none: false
+tokenizer_name: null
+tracker_project_name: train_controlnet_persian
+train_batch_size: 4
+train_data_dir: ./final_dataset
+use_8bit_adam: false
+validation_steps: 1000
+variant: null

logs/train_controlnet_persian/1757354519.00914/events.out.tfevents.1757354519.0f16519829e8.7961.1 ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:97d11b3976c6bfabd0f831a02f62cced3b5b756c39d15929a01b734c0331440a
+size 2355

logs/train_controlnet_persian/1757354519.0106533/hparams.yml ADDED Viewed

+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
+checkpointing_steps: 1000
+checkpoints_total_limit: null
+conditioning_image_column: conditioning_image
+controlnet_model_name_or_path: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: null
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+image_column: image
+learning_rate: 1.0e-05
+logging_dir: logs
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 500
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 90
+mixed_precision: fp16
+num_train_epochs: 15
+num_validation_images: 4
+output_dir: ./controlnet_persian_text_v1
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
+proportion_empty_prompts: 0
+push_to_hub: true
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: null
+set_grads_to_none: false
+tokenizer_name: null
+tracker_project_name: train_controlnet_persian
+train_batch_size: 4
+train_data_dir: ./final_dataset
+use_8bit_adam: false
+validation_steps: 1000
+variant: null

logs/train_controlnet_persian/events.out.tfevents.1757353931.0f16519829e8.4893.0 ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:ab83d9935947b263b52849b9723fb2d936619f76310dd2f8fa2d2f00dde76022
+size 7468

logs/train_controlnet_persian/events.out.tfevents.1757354519.0f16519829e8.7961.0 ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:2962a6ac6c665e781e07000bd6de836eb4e3390f72394ba0635c4e2c7f07f8b6
+size 908588