naumnaum commited on
Commit
affa489
·
verified ·
1 Parent(s): e91520a

Upload 2 files

Browse files
ilvina_aitk_16l_16a_2bs_grad-accum1_0005lr_adamw8bit_1250steps/config.yaml ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: ilvina_aitk_16l_16a_2bs_grad-accum1_0005lr_adamw8bit_1250steps
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: 250
16
+ max_step_saves_to_keep: 1000
17
+ push_to_hub: false
18
+ datasets:
19
+ - folder_path: /workspace/ai-toolkit/datasets/ilvina
20
+ caption_ext: txt
21
+ caption_dropout_rate: 0.05
22
+ shuffle_tokens: false
23
+ cache_latents_to_disk: true
24
+ resolution:
25
+ - 512
26
+ train:
27
+ batch_size: 2
28
+ steps: 1250
29
+ gradient_accumulation_steps: 1
30
+ train_unet: true
31
+ train_text_encoder: false
32
+ gradient_checkpointing: true
33
+ noise_scheduler: flowmatch
34
+ optimizer: adamw8bit
35
+ lr: 0.0005
36
+ lr_scheduler: cosine_with_restarts
37
+ skip_first_sample: true
38
+ linear_timesteps: true
39
+ ema_config:
40
+ use_ema: true
41
+ ema_decay: 0.99
42
+ dtype: bf16
43
+ model:
44
+ name_or_path: black-forest-labs/FLUX.1-dev
45
+ is_flux: true
46
+ quantize: false
47
+ sample:
48
+ sampler: flowmatch
49
+ sample_every: 250
50
+ width: 1024
51
+ height: 1024
52
+ prompts:
53
+ - Portrait of a TOK woman, front-facing, head perfectly centered, solid white
54
+ background
55
+ neg: ''
56
+ seed: 42
57
+ walk_seed: true
58
+ guidance_scale: 4
59
+ sample_steps: 35
60
+ meta:
61
+ name: ilvina_aitk_16l_16a_2bs_grad-accum1_0005lr_adamw8bit_1250steps
62
+ version: '1.0'
ilvina_aitk_16l_16a_2bs_grad-accum1_0005lr_adamw8bit_1250steps/ilvina_aitk_16l_16a_2bs_grad-accum1_0005lr_adamw8bit_1250steps.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd842555564fcf0c66b79747fe8143e53921c3fd188100a02aa093d2ff84b634
3
+ size 171969496