Upload 3 files

test-simpletuner/Training/tr_01/s01_config_01.json

@@ -0,0 +1,67 @@
+{
+    "--model_type": "lora",
+    "--lora_type": "standard",
+    "--pretrained_model_name_or_path": "black-forest-labs/FLUX.1-dev",
+    "--model_family": "flux",
+    "--data_backend_config": "/workspace/s01_multidatabackend.json",
+    "--output_dir": "/workspace/output/test-simpletuner/tr_01/",
+    "--user_prompt_library": "/workspace/s01_prompt_library.json",
+    "--hub_model_id": "simpletuner-lora-test-simpletuner_tr_01",
+    "--tracker_project_name": "simpletuner-lora-test-simpletuner_tr_01",
+    "--tracker_run_name": "tr_01",
+    "--seed": 151851,
+    "--lora_rank": 4,
+    "--lora_alpha": 4,
+    "--mixed_precision": "bf16",
+    "--optimizer": "adamw_bf16",
+    "--learning_rate": "2e-5",
+    "--train_batch_size": 2,
+    "--gradient_accumulation_steps": 2,
+    "--lr_scheduler": "cosine",
+    "--lr_warmup_steps": 100,
+    "--max_train_steps": 30000,
+    "--num_train_epochs": 0,
+    "--checkpointing_steps": 100,
+    "--base_model_precision": "fp8-quanto",
+    "--base_model_default_dtype": "bf16",
+    "--keep_vae_loaded": true,
+    "--flux_lora_target": "all+ffs",
+    "--gradient_precision": "fp32",
+    "--noise_offset": "0.125",
+    "--noise_offset_probability": "0.5",
+    "--checkpoints_total_limit": 20,
+    "--aspect_bucket_rounding": 4,
+    "--minimum_image_size": 0,
+    "--resume_from_checkpoint": "latest",
+    "--push_to_hub": "false",
+    "--push_checkpoints_to_hub": "false",
+    "--report_to": "wandb",
+    "--metadata_update_interval": 60,
+    "--gradient_checkpointing": "true",
+    "--caption_dropout_probability": "0.15",
+    "--resolution_type": "pixel_area",
+    "--resolution": 768,
+    "--validation_seed": 10,
+    "--validation_steps": 100,
+    "--validation_resolution": "512x768",
+    "--validation_guidance": "3.5",
+    "--validation_guidance_rescale": "0.0",
+    "--validation_num_inference_steps": "20",
+    "--validation_prompt": "yellow color",
+    "--num_validation_images": 1,
+    "--validation_torch_compile": "false",
+    "--disable_benchmark": "false",
+    "--snr_gamma": 5,
+    "--inference_scheduler_timestep_spacing": "trailing",
+    "--training_scheduler_timestep_spacing": "trailing",
+    "--max_workers": 32,
+    "--read_batch_size": 25,
+    "--write_batch_size": 64,
+    "--torch_num_threads": 8,
+    "--image_processing_batch_size": 32,
+    "--vae_batch_size": 4,
+    "--compress_disk_cache": true,
+    "--max_grad_norm": "0.04",
+    "--disable_bucket_pruning": false,
+    "--delete_problematic_images": true
+}

test-simpletuner/Training/tr_01/s01_multidatabackend.json

@@ -0,0 +1,65 @@
+[
+    {
+        "id": "all_dataset_768",
+        "type": "local",
+        "instance_data_dir": "/workspace/input/dataset",
+        "crop": false,
+        "crop_style": "random",
+        "crop_aspect": "preserve",
+        "resolution": 768,
+        "resolution_type": "pixel_area",
+        "minimum_image_size": 150,
+        "maximum_image_size": 1024,
+        "target_downsample_size":768,
+        "prepend_instance_prompt": false,
+        "instance_prompt": null,
+        "only_instance_prompt": false,
+        "caption_strategy": "textfile",
+        "cache_dir_vae": "/workspace/cache/cache_vae_768/",
+        "vae_cache_clear_each_epoch": false,
+        "probability": 1.0,
+        "repeats": 3,
+        "disabled": false,
+        "text_embeds": "alt-embed-cache",
+        "skip_file_discovery": "",
+        "preserve_data_backend_cache": false,
+        "metadata_backend": "json",
+        "cache_file_suffix": "_768"
+    },
+    {
+        "id": "all_dataset_512",
+        "type": "local",
+        "instance_data_dir": "/workspace/input/dataset",
+        "crop": false,
+        "crop_style": "random",
+        "crop_aspect": "preserve",
+        "resolution": 512,
+        "resolution_type": "pixel_area",
+        "minimum_image_size": 115,
+        "maximum_image_size": 832,
+        "target_downsample_size": 512,
+        "prepend_instance_prompt": false,
+        "instance_prompt": null,
+        "only_instance_prompt": false,
+        "caption_strategy": "textfile",
+        "cache_dir_vae": "/workspace/cache/cache_vae_512/",
+        "vae_cache_clear_each_epoch": false,
+        "probability": 1.0,
+        "repeats": 3,
+        "disabled": false,
+        "text_embeds": "alt-embed-cache",
+        "skip_file_discovery": "",
+        "preserve_data_backend_cache": false,
+        "metadata_backend": "json",
+        "cache_file_suffix": "_512"
+    },
+    {
+        "id": "alt-embed-cache",
+        "type": "local",
+        "dataset_type": "text_embeds",
+        "default": true,
+        "cache_dir": "/workspace/cache/text_embeds/",
+        "disabled": false,
+        "write_batch_size": 128
+    }
+]

test-simpletuner/Training/tr_01/s01_prompt_library.json

@@ -0,0 +1,6 @@
+{
+    "black": "black color",
+    "green": "green color",
+    "blue": "blue color",
+    "red": "red color"
+}