Samuel96 commited on
Commit
ecd09c3
·
verified ·
1 Parent(s): 726debb

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.yaml +5 -5
  2. lora.safetensors +1 -1
config.yaml CHANGED
@@ -12,7 +12,7 @@ config:
12
  linear_alpha: 16
13
  save:
14
  dtype: float16
15
- save_every: 1001
16
  max_step_saves_to_keep: 1
17
  datasets:
18
  - folder_path: input_images
@@ -26,8 +26,8 @@ config:
26
  - 768
27
  - 1024
28
  train:
29
- batch_size: 2
30
- steps: 1000
31
  gradient_accumulation_steps: 1
32
  train_unet: true
33
  train_text_encoder: false
@@ -35,7 +35,7 @@ config:
35
  gradient_checkpointing: true
36
  noise_scheduler: flowmatch
37
  optimizer: adamw8bit
38
- lr: 0.00045
39
  ema_config:
40
  use_ema: true
41
  ema_decay: 0.99
@@ -46,7 +46,7 @@ config:
46
  quantize: true
47
  sample:
48
  sampler: flowmatch
49
- sample_every: 1001
50
  width: 1024
51
  height: 1024
52
  prompts: []
 
12
  linear_alpha: 16
13
  save:
14
  dtype: float16
15
+ save_every: 201
16
  max_step_saves_to_keep: 1
17
  datasets:
18
  - folder_path: input_images
 
26
  - 768
27
  - 1024
28
  train:
29
+ batch_size: 1
30
+ steps: 200
31
  gradient_accumulation_steps: 1
32
  train_unet: true
33
  train_text_encoder: false
 
35
  gradient_checkpointing: true
36
  noise_scheduler: flowmatch
37
  optimizer: adamw8bit
38
+ lr: 0.0004
39
  ema_config:
40
  use_ema: true
41
  ema_decay: 0.99
 
46
  quantize: true
47
  sample:
48
  sampler: flowmatch
49
+ sample_every: 201
50
  width: 1024
51
  height: 1024
52
  prompts: []
lora.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1068b6529fd679480609bc2a371171e63c4ba7fbb7fbcc2dddbbe2092625f0b6
3
  size 171969408
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd679400daf0ce53ca68c050e7e66bb6c470625a3ea93e65c845bafd5020a7ca
3
  size 171969408