stujlans commited on
Commit
9226a32
1 Parent(s): 7ccaa4b

Upload 4 files

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