Upload folder using huggingface_hub

README.md

@@ -2,4 +2,37 @@
 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: rabbit r1 
+---
+
+# Rabbitr1 Lora
+
+Trained on Replicate using:
+
+https://replicate.com/ostris/flux-dev-lora-trainer/train
+
+
+## Trigger words
+You should use `rabbit r1 ` to trigger the image generation.
+
+
+## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
+
+```py
+from diffusers import AutoPipelineForText2Image
+import torch
+
+pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
+pipeline.load_lora_weights('tourettes/Rabbitr1-lora', weight_name='lora.safetensors')
+image = pipeline('your prompt').images[0]
+```
+
+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

@@ -1,60 +1,59 @@
+job: custom_job
 config:
  name: flux_train_replicate
  process:
- - datasets:
- - cache_latents_to_disk: true
+ - type: custom_sd_trainer
+ training_folder: output
+ device: cuda:0
+ trigger_word: 'rabbit r1 '
+ network:
+ type: lora
+ linear: 128
+ linear_alpha: 128
+ save:
+ dtype: float16
+ save_every: 2001
+ max_step_saves_to_keep: 1
+ datasets:
+ - folder_path: input_images
+ caption_ext: txt
  caption_dropout_rate: 0.05
- caption_ext: filename
- folder_path: input_images
+ shuffle_tokens: false
+ cache_latents_to_disk: true
  resolution:
  - 512
  - 768
  - 1024
- shuffle_tokens: false
- device: cuda:0
+ train:
+ batch_size: 1
+ steps: 2000
+ gradient_accumulation_steps: 1
+ train_unet: true
+ train_text_encoder: false
+ content_or_style: balanced
+ gradient_checkpointing: true
+ noise_scheduler: flowmatch
+ optimizer: adamw8bit
+ lr: 0.0004
+ ema_config:
+ use_ema: true
+ ema_decay: 0.99
+ dtype: bf16
  model:
+ name_or_path: FLUX.1-dev
  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
+ sampler: flowmatch
+ sample_every: 2001
+ width: 1024
  height: 1024
+ prompts: []
  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: 1001
- 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: 1000
- train_text_encoder: false
- train_unet: true
- training_folder: output
- trigger_word: TOK
- type: sd_trainer
-job: extension
+ guidance_scale: 4
+ sample_steps: 20
 meta:
  name: flux_train_replicate
  version: '1.0'

lora.safetensors

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