Upload folder using huggingface_hub

README.md

@@ -2,47 +2,4 @@
 license: other
 license_name: flux-1-dev-non-commercial-license
 license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
-language:
-- en
-tags:
-- flux
-- diffusers
-- lora
-base_model: "black-forest-labs/FLUX.1-dev"
-pipeline_tag: text-to-image
-instance_prompt: ser
----
-
-# ser LoRa Trainer (developing...)
-
-Trained on Replicate using:
-
-https://replicate.com/ostris/flux-dev-lora-trainer/train
-
-
-## Trigger words
-You should use `mert` to trigger the image generation.
-
-
-## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
-
-```py
-import torch
-from diffusers import FluxPipeline
-
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
-pipe.load_lora_weights('codermert/ayd_flux', weight_name='flux_train_replicate.safetensors')
-pipe.fuse_lora(lora_scale=0.9)
-pipe.to("cuda")
-
-prompt = "close up portrait, Amidst the interplay of light and shadows in a photography studio,a soft spotlight traces the contours of a face,highlighting a figure clad in a sleek black turtleneck. The garment,hugging the skin with subtle luxury,complements the Caucasian model's understated makeup,embodying minimalist elegance. Behind,a pale gray backdrop extends,its fine texture shimmering subtly in the dim light,artfully balancing the composition and focusing attention on the subject. In a palette of black,gray,and skin tones,simplicity intertwines with profundity,as every detail whispers untold stories."
-
-image = pipe(prompt, 
-             num_inference_steps=24, 
-             guidance_scale=3.5,
-             width=768, height=1024,
-            ).images[0]
-image.save(f"example.png")
-```
-
-For more details, including weighting, merging and fusing LoRAs, check the [documentation on loading LoRAs in diffusers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters)
+---

config.yaml

@@ -0,0 +1,60 @@
+config:
+  name: flux_train_replicate
+  process:
+  - datasets:
+    - cache_latents_to_disk: true
+      caption_dropout_rate: 0.05
+      caption_ext: txt
+      folder_path: input_images
+      resolution:
+      - 512
+      - 768
+      - 1024
+      shuffle_tokens: false
+    device: cuda:0
+    model:
+      is_flux: true
+      name_or_path: black-forest-labs/FLUX.1-dev
+      quantize: true
+    network:
+      linear: 16
+      linear_alpha: 16
+      type: lora
+    sample:
+      guidance_scale: 4
+      height: 1024
+      neg: ''
+      prompts:
+      - a sign that says 'I LOVE PROMPTS!' in the style of [trigger]
+      sample_every: 250
+      sample_steps: 20
+      sampler: flowmatch
+      seed: 42
+      walk_seed: true
+      width: 1024
+    save:
+      dtype: float16
+      max_step_saves_to_keep: 1
+      save_every: 1501
+    train:
+      batch_size: 1
+      content_or_style: balanced
+      dtype: bf16
+      ema_config:
+        ema_decay: 0.99
+        use_ema: true
+      gradient_accumulation_steps: 1
+      gradient_checkpointing: true
+      lr: 0.0004
+      noise_scheduler: flowmatch
+      optimizer: adamw8bit
+      steps: 1500
+      train_text_encoder: false
+      train_unet: true
+    training_folder: output
+    trigger_word: TOK
+    type: sd_trainer
+job: extension
+meta:
+  name: flux_train_replicate
+  version: '1.0'

flux_train_replicate.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca3aaaab20a9213da9d059f45399c2b4e6ed90025423bd18a5ede73a7e27fb24
+size 171969416

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:12c3a93498dcb03148444df6bfc5c5f0fd069f4e793648829568b746f7eb9676
+size 173272836