Fix: revert back to previous version of the model since the new training failed

config.yaml

@@ -12,22 +12,21 @@ config:
       linear_alpha: 16
     save:
       dtype: float16
-      save_every: 1601
+      save_every: 1001
       max_step_saves_to_keep: 1
     datasets:
     - folder_path: input_images
       caption_ext: txt
-      caption_dropout_rate: 0.2
+      caption_dropout_rate: 0.05
       shuffle_tokens: false
-      cache_latents_to_disk: false
-      cache_latents: true
+      cache_latents_to_disk: true
       resolution:
       - 512
       - 768
       - 1024
     train:
-      batch_size: 5
-      steps: 1600
+      batch_size: 1
+      steps: 1000
       gradient_accumulation_steps: 1
       train_unet: true
       train_text_encoder: false
@@ -46,15 +45,15 @@ config:
       quantize: true
     sample:
       sampler: flowmatch
-      sample_every: 1601
+      sample_every: 1001
       width: 1024
       height: 1024
       prompts: []
       neg: ''
       seed: 42
       walk_seed: true
-      guidance_scale: 3.5
-      sample_steps: 28
+      guidance_scale: 4
+      sample_steps: 20
 meta:
   name: flux_train_replicate
   version: '1.0'

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6b7a3d6bf24ba01bdc4eda8988f13de91dd94a9af03e15b72e4792c5059ddb84
-size 171969424
+oid sha256:41010aee95a4b3411b2a4a74873c78c5dac62eb1ce473307c68140f38f39b8d1
+size 171969416