Upload folder using huggingface_hub

dataset.toml

@@ -9,6 +9,6 @@ batch_size = 1
 keep_tokens = 1
 
   [[datasets.subsets]]
-  image_dir = '/app/fluxgym/datasets/sarahrunpod'
-  class_tokens = 'sarahrunpod'
+  image_dir = '/app/fluxgym/datasets/sarahrunpod2'
+  class_tokens = 'sarahrunpod2'
   num_repeats = 10

sample_prompts.txt

@@ -1 +1 @@
-sarahrunpod
+sarahrunpod2

sarahrunpod2-000005.safetensors

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

sarahrunpod2-000010.safetensors

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

sarahrunpod2.safetensors

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

train.sh

@@ -15,18 +15,18 @@ accelerate launch \
   --mixed_precision bf16 \
   --save_precision bf16 \
   --network_module networks.lora_flux \
-  --network_dim 4 \
+  --network_dim 32 \
   --optimizer_type adamw8bit \
-  --learning_rate 8e-4 \
+  --learning_rate 4e-4 \
   --cache_text_encoder_outputs \
   --cache_text_encoder_outputs_to_disk \
   --fp8_base \
   --highvram \
-  --max_train_epochs 16 \
-  --save_every_n_epochs 4 \
-  --dataset_config "/app/fluxgym/outputs/sarahrunpod/dataset.toml" \
-  --output_dir "/app/fluxgym/outputs/sarahrunpod" \
-  --output_name sarahrunpod \
+  --max_train_epochs 12 \
+  --save_every_n_epochs 5 \
+  --dataset_config "/app/fluxgym/outputs/sarahrunpod2/dataset.toml" \
+  --output_dir "/app/fluxgym/outputs/sarahrunpod2" \
+  --output_name sarahrunpod2 \
   --timestep_sampling shift \
   --discrete_flow_shift 3.1582 \
   --model_prediction_type raw \