fofr commited on
Commit
4a405fa
1 Parent(s): ef346c2

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +14 -0
  2. config.yaml +58 -0
  3. lora.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ license_name: flux-1-dev-non-commercial-license
4
+ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
5
+ language:
6
+ - en
7
+ tags:
8
+ - flux
9
+ base_model: "black-forest-labs/FLUX.1-dev"
10
+ pipeline_tag: text-to-image
11
+ ---
12
+
13
+ Trained on Replicate using:
14
+ https://replicate.com/ostris/flux-dev-lora-trainer/train
config.yaml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: flux_train_replicate
4
+ process:
5
+ - type: sd_trainer
6
+ training_folder: output
7
+ device: cuda:0
8
+ network:
9
+ type: lora
10
+ linear: 16
11
+ linear_alpha: 16
12
+ save:
13
+ dtype: float16
14
+ save_every: 1001
15
+ max_step_saves_to_keep: 1
16
+ datasets:
17
+ - folder_path: input_images
18
+ caption_ext: filename
19
+ caption_dropout_rate: 0.05
20
+ shuffle_tokens: false
21
+ cache_latents_to_disk: true
22
+ resolution:
23
+ - 512
24
+ - 768
25
+ - 1024
26
+ train:
27
+ batch_size: 1
28
+ steps: 1000
29
+ gradient_accumulation_steps: 1
30
+ train_unet: true
31
+ train_text_encoder: false
32
+ content_or_style: balanced
33
+ gradient_checkpointing: true
34
+ noise_scheduler: flowmatch
35
+ optimizer: adamw8bit
36
+ lr: 0.0004
37
+ ema_config:
38
+ use_ema: true
39
+ ema_decay: 0.99
40
+ dtype: bf16
41
+ model:
42
+ name_or_path: FLUX.1-dev
43
+ is_flux: true
44
+ quantize: true
45
+ sample:
46
+ sampler: flowmatch
47
+ sample_every: 1001
48
+ width: 1024
49
+ height: 1024
50
+ prompts: []
51
+ neg: ''
52
+ seed: 42
53
+ walk_seed: true
54
+ guidance_scale: 4
55
+ sample_steps: 20
56
+ meta:
57
+ name: flux_train_replicate
58
+ version: '1.0'
lora.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f1cf7143cd63331b259923090afd171aa456149bf387996eaf5837683842d1c
3
+ size 171969368