StonedApe commited on
Commit
920ee56
1 Parent(s): 09ef3d2

Upload 4 files

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