Upload folder using huggingface_hub

config.yaml

@@ -8,16 +8,16 @@ config:
     trigger_word: TOK
     network:
       type: lora
-      linear: 16
-      linear_alpha: 16
+      linear: 24
+      linear_alpha: 24
     save:
       dtype: float16
-      save_every: 51
+      save_every: 501
       max_step_saves_to_keep: 1
     datasets:
     - folder_path: input_images
       caption_ext: txt
-      caption_dropout_rate: 0.05
+      caption_dropout_rate: 0.1
       shuffle_tokens: false
       cache_latents_to_disk: true
       resolution:
@@ -25,8 +25,8 @@ config:
       - 768
       - 1024
     train:
-      batch_size: 1
-      steps: 50
+      batch_size: 4
+      steps: 500
       gradient_accumulation_steps: 1
       train_unet: true
       train_text_encoder: false
@@ -34,7 +34,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
@@ -45,7 +45,7 @@ config:
       quantize: true
     sample:
       sampler: flowmatch
-      sample_every: 51
+      sample_every: 501
       width: 1024
       height: 1024
       prompts: []

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6813878fd5e29ebfb8d58d4cc0b2525936b6f819da87e959afc07eb0319a11c9
-size 171969408
+oid sha256:66498e9dca31fb27b14ccb05e9dd1611d44a88d5d90783e4fa9d99caef1ff814
+size 257887536