charlesai commited on
Commit
5d76f7d
1 Parent(s): 3d5dedf

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.yaml +60 -0
  2. flux_train_replicate.safetensors +3 -0
  3. optimizer.pt +3 -0
config.yaml ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ config:
2
+ name: flux_train_replicate
3
+ process:
4
+ - datasets:
5
+ - cache_latents_to_disk: true
6
+ caption_dropout_rate: 0.05
7
+ caption_ext: filename
8
+ folder_path: input_images
9
+ resolution:
10
+ - 512
11
+ - 768
12
+ - 1024
13
+ shuffle_tokens: false
14
+ device: cuda:0
15
+ model:
16
+ is_flux: true
17
+ name_or_path: black-forest-labs/FLUX.1-dev
18
+ quantize: true
19
+ network:
20
+ linear: 16
21
+ linear_alpha: 16
22
+ type: lora
23
+ sample:
24
+ guidance_scale: 4
25
+ height: 1024
26
+ neg: ''
27
+ prompts:
28
+ - a sign that says 'I LOVE PROMPTS!' in the style of [trigger]
29
+ sample_every: 250
30
+ sample_steps: 20
31
+ sampler: flowmatch
32
+ seed: 42
33
+ walk_seed: true
34
+ width: 1024
35
+ save:
36
+ dtype: float16
37
+ max_step_saves_to_keep: 1
38
+ save_every: 4001
39
+ train:
40
+ batch_size: 1
41
+ content_or_style: balanced
42
+ dtype: bf16
43
+ ema_config:
44
+ ema_decay: 0.99
45
+ use_ema: true
46
+ gradient_accumulation_steps: 1
47
+ gradient_checkpointing: true
48
+ lr: 0.0004
49
+ noise_scheduler: flowmatch
50
+ optimizer: adamw8bit
51
+ steps: 4000
52
+ train_text_encoder: false
53
+ train_unet: true
54
+ training_folder: output
55
+ trigger_word: TOK
56
+ type: sd_trainer
57
+ job: extension
58
+ meta:
59
+ name: flux_train_replicate
60
+ version: '1.0'
flux_train_replicate.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8842948c91ae9d8f23d1227922d73fd45704def9c97d86c3c83997f870ee3950
3
+ size 171969408
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4388c9fb6744c50b75a9737a5c7227348e6ba65d9ed1307e833f9c703ca1e17a
3
+ size 173272836