Upload folder using huggingface_hub

README.md

@@ -16,7 +16,7 @@ pipeline_tag: text-to-image
 #       prompt
 #     output:
 #       url: https://...
-instance_prompt: TOKEN
+instance_prompt: TOK
 ---
 
 # 123Test
@@ -29,7 +29,7 @@ https://replicate.com/ostris/flux-dev-lora-trainer/train
 
 
 ## Trigger words
-You should use `TOKEN` to trigger the image generation.
+You should use `TOK` to trigger the image generation.
 
 
 ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)

config.yaml

@@ -5,14 +5,14 @@ config:
   - type: custom_sd_trainer
     training_folder: output
     device: cuda:0
-    trigger_word: TOKEN
+    trigger_word: TOK
     network:
       type: lora
       linear: 16
       linear_alpha: 16
     save:
       dtype: float16
-      save_every: 51
+      save_every: 1001
       max_step_saves_to_keep: 1
     datasets:
     - folder_path: input_images
@@ -23,9 +23,11 @@ config:
       cache_latents: true
       resolution:
       - 512
+      - 768
+      - 1024
     train:
-      batch_size: 1
-      steps: 50
+      batch_size: 2
+      steps: 1000
       gradient_accumulation_steps: 1
       train_unet: true
       train_text_encoder: false
@@ -33,7 +35,7 @@ config:
       gradient_checkpointing: true
       noise_scheduler: flowmatch
       optimizer: adamw8bit
-      lr: 0.0004
+      lr: 0.00045
       ema_config:
         use_ema: true
         ema_decay: 0.99
@@ -44,7 +46,7 @@ config:
       quantize: true
     sample:
       sampler: flowmatch
-      sample_every: 51
+      sample_every: 1001
       width: 1024
       height: 1024
       prompts: []

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4dc257d75091930d54676003e9360cb85338ae91a16fe2207b280b9fd1efec86
-size 171969416
+oid sha256:1068b6529fd679480609bc2a371171e63c4ba7fbb7fbcc2dddbbe2092625f0b6
+size 171969408