upload config

br-config.env

@@ -0,0 +1,59 @@
+export MODEL_TYPE='lora'
+export TRAINING_SEED=420
+
+export CHECKPOINTING_STEPS=500
+export CHECKPOINTING_LIMIT=10
+
+export LEARNING_RATE=1e-4
+
+export FLUX=true
+export MODEL_NAME="black-forest-labs/FLUX.1-dev"
+
+export VALIDATION_SEED=420
+export VALIDATION_PROMPT="a photograph of Brandon Rowland"
+export VALIDATION_NEGATIVE_PROMPT=""
+
+# How frequently we will save and run a pipeline for validations.
+export VALIDATION_STEPS=500
+
+# Validation image settings.
+export VALIDATION_GUIDANCE=3.5
+export VALIDATION_GUIDANCE_REAL=1.0
+export VALIDATION_NUM_INFERENCE_STEPS=28
+export VALIDATION_GUIDANCE_RESCALE=0
+export VALIDATION_RESOLUTION=1024x1024,832x1216
+
+export ALLOW_TF32=true
+export PURE_BF16=true
+
+export CAPTION_DROPOUT_PROBABILITY=0
+
+export MAX_NUM_STEPS=0
+export NUM_EPOCHS=1000
+
+export OPTIMIZER="adamw_bf16"
+export LR_SCHEDULE="constant"
+export LR_WARMUP_STEPS=500
+
+export TRAIN_BATCH_SIZE=1
+
+export RESOLUTION=512
+export RESOLUTION_TYPE=pixel
+
+export GRADIENT_ACCUMULATION_STEPS=2
+export MIXED_PRECISION="bf16"
+export TRAINING_DYNAMO_BACKEND='no'
+
+export USE_XFORMERS=false
+export USE_GRADIENT_CHECKPOINTING=true
+export VAE_BATCH_SIZE=8
+export TRAINER_EXTRA_ARGS="--max_grad_norm=1.0 --gradient_precision=fp32 --base_model_default_dtype=bf16 --lora_init_type=default --flux_lora_target=all+ffs --flux_guidance_value=1 --user_prompt_library=user_prompt_library.json"
+
+export USE_EMA=false
+
+export PUSH_TO_HUB="true"
+export PUSH_CHECKPOINTS="true"
+export HUB_MODEL_NAME="brandon-rowland-flux"
+
+export DATALOADER_CONFIG="config/br-multidatabackend.json"
+export OUTPUT_DIR="output-br/models"

br-multidatabackend.json

@@ -0,0 +1,30 @@
+[
+  {
+    "id": "Brandon",
+    "type": "local",
+    "crop": false,
+    "crop_aspect": "square",
+    "crop_style": "center",
+    "resolution": 512,
+    "minimum_image_size": 512,
+    "maximum_image_size": 768,
+    "target_downsample_size": 512,
+    "resolution_type": "pixel",
+    "cache_dir_vae": "cache/vae/br",
+    "instance_data_dir": "datasets/brandon-rowland",
+    "disabled": false,
+    "skip_file_discovery": "",
+    "caption_strategy": "textfile",
+    "metadata_backend": "json",
+    "repeats": 0
+  },
+  {
+    "id": "text-embeds",
+    "type": "local",
+    "dataset_type": "text_embeds",
+    "default": true,
+    "cache_dir": "cache/text/flux/br",
+    "disabled": false,
+    "write_batch_size": 128
+  }
+]

user_prompt_library.json

@@ -0,0 +1,5 @@
+{
+    "1": "a photo of Brandon Rowland",
+    "2": "a photo of Brandon Rowland leaning against a brick wall, arms crossed, wearing a white tank top and ripped jeans. His gaze is intense, directed at the camera. The background includes a graffiti-covered wall and an urban street scene",
+	"both": "A photo of Brandon Rowland. Shirtless, lying on a dark blue sofa, one hand resting behind his head, the other on his chest. The background shows a bookshelf filled with books and plants. The lighting is dim, creating a cozy evening ambiance"
+}