Upload CN_ControlBody_SD_v1_diffusers_train_v1.json

CN_ControlBody_SD_v1_diffusers_train_v1.json

@@ -0,0 +1,89 @@
+{
+  "config_save_dir": "output/configs",
+  "mixed_precision_type": "no",
+  "num_processes": 12,
+  "num_machines": 1,
+  "num_cpu_threads_per_process": 4,
+  "dynamo_backend": "no",
+  "multi_gpu": false,
+  "gpu_ids": "",
+  "main_process_port": 0,
+  "extra_args": "",
+  "pretrained_model_name_or_path": "/cn_train/waiNSFWIllustrious_v80_VRoidstyle",
+  "pretrained_vae_model_name_or_path": "madebyollin/sdxl-vae-fp16-fix",
+  "controlnet_model_name_or_path": "",
+  "variant": "",
+  "revision": "",
+  "tokenizer_name": "",
+  "output_dir": "/workspace/cn_output/2025-04-11",
+  "cache_dir": "/workspace/cn_output/cache",
+  "dataset_name": "Visual-Bank/CN_ControlBody_SD_v1_diffusers",
+  "train_data_dir": "",
+  "dataset_config_name": "",
+  "dataloader_num_workers": 0,
+  "image_column": "image",
+  "conditioning_image_column": "conditioning_image",
+  "caption_column": "text",
+  "max_train_samples": null,
+  "proportion_empty_prompts": 0,
+  "train_batch_size": 2,
+  "max_train_steps": 50000,
+  "num_train_epochs": null,
+  "seed": 42,
+  "resolution": 1024,
+  "crops_coords_top_left_h": 0,
+  "crops_coords_top_left_w": 0,
+  "learning_rate": 1e-05,
+  "scale_lr": false,
+  "lr_scheduler": "cosine",
+  "lr_warmup_steps": 500,
+  "lr_num_cycles": 1,
+  "lr_power": 1,
+  "gradient_accumulation_steps": 1,
+  "gradient_checkpointing": false,
+  "use_8bit_adam": false,
+  "adam_beta1": 0.9,
+  "adam_beta2": 0.999,
+  "adam_weight_decay": 0.01,
+  "adam_epsilon": 1e-08,
+  "max_grad_norm": 1,
+  "set_grads_to_none": false,
+  "checkpointing_steps": 5000,
+  "checkpoints_total_limit": null,
+  "resume_from_checkpoint": "",
+  "num_validation_images": 2,
+  "validation_steps": 1000,
+  "validation_prompt": [
+    "3D, 1girl, solo, white background, simple background, green shorts, from behind, green pants, brown hair, shoes, sweater, long sleeves, red sweater, short hair, black footwear, shorts, hands up, pants, standing on one leg, facing away, standing",
+    "3D, 1girl, solo, brown eyes, brown hair, white background, bangs, simple background, open mouth, chibi, upper teeth only, full body, long sleeves, pants, white footwear, teeth, green pants, green shorts, smile, shirt, shoes, blunt bangs, sweater, shorts",
+    "3D, 1girl, solo, brown hair, brown eyes, white background, simple background, standing on one leg, green pants, pants, sandals, standing, full body, long sleeves, open mouth, shirt, sweater, looking at viewer, bangs, upper teeth only, teeth, green shorts, white footwear",
+    "3D, 1girl, solo, blue hair, long hair, chibi, white background, black footwear, purple eyes, boots, bangs, necktie, sitting, black necktie, twintails, simple background, from side, long sleeves, black pants, pantyhose, pants, full body, shirt, low twintails",
+    "3D, 1girl, solo, long hair, green eyes, skirt, brown hair, white background, shirt, black skirt, simple background, black footwear, white shirt, long sleeves, full body, standing, profile, shoes",
+    "3D, 1girl, long hair, solo, white background, brown hair, very long hair, simple background, facing away, long sleeves, sandals, from behind, squatting, dress, shirt, full body, black dress, black skirt, standing",
+    "3D, 1girl, solo, green eyes, long hair, skirt, brown hair, white background, black skirt, full body, shirt, white shirt, simple background, pointing, sandals, long sleeves, white footwear, bangs, crop top",
+    "3D, 1girl, solo, shirt, green hair, pants, white shirt, plaid, plaid pants, glasses, bow, white background, full body, simple background, shoes, bowtie, open mouth, brown footwear, short sleeves, bangs, collared shirt, hands up, purple eyes, black bow, standing on one leg, standing, short hair, smile, looking at viewer, red-framed eyewear, green eyes, dress shirt, school uniform",
+    "3D, 1girl, solo, green hair, plaid pants, glasses, pants, shirt, chibi, white shirt, plaid, white background, purple eyes, green pants, looking at viewer, full body, simple background, brown footwear, shoes, arms up, bangs, closed mouth, kneeling",
+    "3D, 1girl, solo, green hair, shirt, pants, white shirt, plaid pants, plaid, glasses, short hair, white background, bow, bowtie, full body, shoes, simple background, short sleeves, gumi, arm up, standing, brown footwear, green pants, smile, bangs, closed mouth"
+  ],
+  "validation_image": [
+    "/workspace/test_img/000005.png",
+    "/workspace/test_img/000235.png",
+    "/workspace/test_img/000247.png",
+    "/workspace/test_img/008176.png",
+    "/workspace/test_img/014307.png",
+    "/workspace/test_img/014318.png",
+    "/workspace/test_img/014386.png",
+    "/workspace/test_img/023672.png",
+    "/workspace/test_img/023684.png",
+    "/workspace/test_img/023728.png"
+  ],
+  "report_to": "tensorboard",
+  "logging_dir": "logs",
+  "tracker_project_name": "CN_ControlBody_SD_v1_diffusers_train_v1",
+  "push_to_hub": true,
+  "hub_model_id": "Visual-Bank/CN_ControlBody_SD_v1_diffusers_train_v1",
+  "hub_token": "",
+  "allow_tf32": false,
+  "enable_xformers_memory_efficient_attention": true,
+  "enable_npu_flash_attention": false
+}