Upload folder using huggingface_hub

README.md

@@ -11,11 +11,18 @@ tags:
 - replicate
 base_model: "black-forest-labs/FLUX.1-dev"
 pipeline_tag: text-to-image
+# widget:
+#   - text: >-
+#       prompt
+#     output:
+#       url: https://...
 instance_prompt: TOK
 ---
 
 # Flux Custom Smaller
 
+<!-- <Gallery /> -->
+
 Trained on Replicate using:
 
 https://replicate.com/ostris/flux-dev-lora-trainer/train

config.yaml

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

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:08759a242a6b6bb8ca03f7bda015e25de9300bd4995b9ba5d74bf2cfd5c878bb
-size 171969416
+oid sha256:70130a2b7819e8a53ad640ea18bba5ce6c8f5136cf516ca0e3d7012790e7926e
+size 171969408