First model run 10k iterations

args.json

@@ -1,5 +1,5 @@
 {
-  "pretrained_model_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/X000",
+  "pretrained_model_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/2000",
   "pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
   "revision": null,
   "tokenizer_name": null,
@@ -15,7 +15,7 @@
   "pad_tokens": false,
   "with_prior_preservation": true,
   "prior_loss_weight": 1.0,
-  "num_class_images": 50,
+  "num_class_images": 100,
   "output_dir": "/content/drive/MyDrive/demslay_diffusion_weights/zwx",
   "seed": 1337,
   "resolution": 512,
@@ -23,11 +23,11 @@
   "train_text_encoder": true,
   "train_batch_size": 1,
   "sample_batch_size": 4,
-  "num_train_epochs": 60,
-  "max_train_steps": 3000,
+  "num_train_epochs": 10,
+  "max_train_steps": 1000,
   "gradient_accumulation_steps": 1,
   "gradient_checkpointing": false,
-  "learning_rate": 5e-06,
+  "learning_rate": 1e-06,
   "scale_lr": false,
   "lr_scheduler": "polynomial",
   "lr_warmup_steps": 0,
@@ -42,7 +42,7 @@
   "hub_model_id": null,
   "logging_dir": "logs",
   "log_interval": 10,
-  "save_interval": 1000,
+  "save_interval": 2000,
   "save_min_steps": 0,
   "mixed_precision": "fp16",
   "not_cache_latents": false,
@@ -51,9 +51,9 @@
   "concepts_list": [
     {
       "instance_prompt": "demslay style",
-      "class_prompt": "style",
+      "class_prompt": "person",
       "instance_data_dir": "/content/data/demslay",
-      "class_data_dir": "/content/data/style"
+      "class_data_dir": "/content/data/person"
     }
   ]
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/X000",
+  "_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/2000",
   "architectures": [
     "CLIPTextModel"
   ],

text_encoder/pytorch_model.bin

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

tokenizer/tokenizer_config.json

@@ -19,7 +19,7 @@
   },
   "errors": "replace",
   "model_max_length": 77,
-  "name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/X000/tokenizer",
+  "name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/2000/tokenizer",
   "pad_token": "<|endoftext|>",
   "special_tokens_map_file": "./special_tokens_map.json",
   "tokenizer_class": "CLIPTokenizer",

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.9.0",
-  "_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/X000",
+  "_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/2000",
   "act_fn": "silu",
   "attention_head_dim": 8,
   "block_out_channels": [

unet/diffusion_pytorch_model.bin

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

vae/diffusion_pytorch_model.bin

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