Upload folder using huggingface_hub

README.md

@@ -11,7 +11,7 @@ tags:
 - replicate
 base_model: "black-forest-labs/FLUX.1-dev"
 pipeline_tag: text-to-image
-instance_prompt: k1arnacard
+instance_prompt: kcblack
 ---
 
 # Martin Flux 1
@@ -22,7 +22,7 @@ https://replicate.com/ostris/flux-dev-lora-trainer/train
 
 
 ## Trigger words
-You should use `k1arnacard` to trigger the image generation.
+You should use `kcblack` 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: k1arnacard
+    trigger_word: kcblack
     network:
       type: lora
-      linear: 20
-      linear_alpha: 20
+      linear: 16
+      linear_alpha: 16
     save:
       dtype: float16
-      save_every: 1301
+      save_every: 1001
       max_step_saves_to_keep: 1
     datasets:
     - folder_path: input_images
@@ -26,15 +26,15 @@ config:
       - 1024
     train:
       batch_size: 1
-      steps: 1300
+      steps: 1000
       gradient_accumulation_steps: 1
       train_unet: true
       train_text_encoder: false
       content_or_style: balanced
       gradient_checkpointing: true
       noise_scheduler: flowmatch
-      optimizer: prodigy
-      lr: 0.0003
+      optimizer: adamw8bit
+      lr: 0.0004
       ema_config:
         use_ema: true
         ema_decay: 0.99
@@ -45,7 +45,7 @@ config:
       quantize: true
     sample:
       sampler: flowmatch
-      sample_every: 1301
+      sample_every: 1001
       width: 1024
       height: 1024
       prompts: []

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6e79e72b7e1b88c7d8afdcf1b93a0c73bd4135ad09cb69bb42e52e768582fb12
-size 214928528
+oid sha256:2f86921898a37f17f83ba95d15308764c653195a2d01e743014425cafe2d98db
+size 171969424