NikoKZ commited on
Commit
b75c470
·
verified ·
1 Parent(s): 5ae49fc

Upload 4 files

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