boisterous commited on
Commit
70f2f15
·
verified ·
1 Parent(s): 01fdc54

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +7 -7
config.yaml CHANGED
@@ -5,14 +5,14 @@ config:
5
  - type: custom_sd_trainer
6
  training_folder: output
7
  device: cuda:0
8
- trigger_word: OaiT
9
  network:
10
  type: lora
11
- linear: 16
12
- linear_alpha: 16
13
  save:
14
  dtype: float16
15
- save_every: 1501
16
  max_step_saves_to_keep: 1
17
  datasets:
18
  - folder_path: input_images
@@ -27,7 +27,7 @@ config:
27
  - 1024
28
  train:
29
  batch_size: 1
30
- steps: 1500
31
  gradient_accumulation_steps: 1
32
  train_unet: true
33
  train_text_encoder: false
@@ -46,7 +46,7 @@ config:
46
  quantize: true
47
  sample:
48
  sampler: flowmatch
49
- sample_every: 1501
50
  width: 1024
51
  height: 1024
52
  prompts: []
@@ -57,4 +57,4 @@ config:
57
  sample_steps: 28
58
  meta:
59
  name: flux_train_replicate
60
- version: '1.0'
 
5
  - type: custom_sd_trainer
6
  training_folder: output
7
  device: cuda:0
8
+ trigger_word: OiaT
9
  network:
10
  type: lora
11
+ linear: 20
12
+ linear_alpha: 20
13
  save:
14
  dtype: float16
15
+ save_every: 2001
16
  max_step_saves_to_keep: 1
17
  datasets:
18
  - folder_path: input_images
 
27
  - 1024
28
  train:
29
  batch_size: 1
30
+ steps: 2000
31
  gradient_accumulation_steps: 1
32
  train_unet: true
33
  train_text_encoder: false
 
46
  quantize: true
47
  sample:
48
  sampler: flowmatch
49
+ sample_every: 2001
50
  width: 1024
51
  height: 1024
52
  prompts: []
 
57
  sample_steps: 28
58
  meta:
59
  name: flux_train_replicate
60
+ version: '1.0'