Additional 8k training iter

args.json

@@ -1,5 +1,5 @@
 {
-  "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "pretrained_model_name_or_path": "MHesho/pachuto-diffusion",
   "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": 1000,
+  "num_class_images": 2000,
   "output_dir": "/content/drive/MyDrive/pachutorresart_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": 8,
+  "num_train_epochs": 4,
   "max_train_steps": 8000,
   "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,
@@ -51,9 +51,9 @@
   "concepts_list": [
     {
       "instance_prompt": "pachutorresart style",
-      "class_prompt": "illustration style",
+      "class_prompt": "style",
       "instance_data_dir": "/content/data/pachutorresart",
-      "class_data_dir": "/content/data/illustration style"
+      "class_data_dir": "/content/data/style"
     }
   ]
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "_name_or_path": "MHesho/pachuto-diffusion",
   "architectures": [
     "CLIPTextModel"
   ],

text_encoder/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:50895b2c64795985e2c1cbe9d85faa197718314a8810f80963c60ab296c6a8f5
+oid sha256:0e3cdf83e3f54542c056d978c1dd64a9b63f639d7aa9720777d21f495d8150d6
 size 492309793

tokenizer/tokenizer_config.json

@@ -19,7 +19,7 @@
   },
   "errors": "replace",
   "model_max_length": 77,
-  "name_or_path": "/root/.cache/huggingface/diffusers/models--runwayml--stable-diffusion-v1-5/snapshots/889b629140e71758e1e0006e355c331a5744b4bf/tokenizer",
+  "name_or_path": "/root/.cache/huggingface/diffusers/models--MHesho--pachuto-diffusion/snapshots/2400c90c42fef3e8418ca186a55bfbf4a0ee78bf/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": "runwayml/stable-diffusion-v1-5",
+  "_name_or_path": "MHesho/pachuto-diffusion",
   "act_fn": "silu",
   "attention_head_dim": 8,
   "block_out_channels": [

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:949cbbab97002c66e650ed5816be47157489f55c9a59803a06ef193662acf9c9
+oid sha256:a80ac4686dc6ce976e97fd0c16616ab7f73d4c1a6f56080fb8aebac2f8129fd5
 size 3438375973