Upload folder using huggingface_hub

README.md

@@ -21,6 +21,9 @@ Trained on Replicate using:
 https://replicate.com/ostris/flux-dev-lora-trainer/train
 
 
+## Trigger words
+You should use `TOK` to trigger the image generation.
+
 
 ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
 

config.yaml

@@ -12,7 +12,7 @@ config:
       linear_alpha: 16
     save:
       dtype: float16
-      save_every: 1001
+      save_every: 4001
       max_step_saves_to_keep: 1
     datasets:
     - folder_path: input_images
@@ -26,7 +26,7 @@ config:
       - 1024
     train:
       batch_size: 1
-      steps: 1000
+      steps: 4000
       gradient_accumulation_steps: 1
       train_unet: true
       train_text_encoder: false
@@ -45,7 +45,7 @@ config:
       quantize: true
     sample:
       sampler: flowmatch
-      sample_every: 1001
+      sample_every: 4001
       width: 1024
       height: 1024
       prompts: []

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:89f38e73059c8a7176034fb798cc4d4034fc8e82533a62f98366a946d3e5d45e
+oid sha256:9c26fb8fbd6738996d8e601561f810b6a2616471585f9aa5c42c1d41a7afa081
 size 171969408